zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-28 15:08:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix index calculation

Browse Source
This commit is contained in:
Chime Ogbuji 2024-11-03 20:36:55 -05:00 committed by Awni Hannun
parent a4a86ad898
commit a5b866cf73
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mlx_lm/tuner/datasets.py
View File

@ -97,7 +97,7 @@ class CompletionsDatasetCollection:
while True: while True:
try: try:
if (curr_idx + 1) < len(item): if (curr_idx + 1) <= len(item):
return item[curr_idx] return item[curr_idx]
else: else:
curr_idx -= len(item) curr_idx -= len(item)