zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-30 02:53:41 +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
parent 24f40c3b8d
commit 78b24a2375
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

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