ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2024-12-26 22:25:19 -08:00
commit 52b24512de
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/_utils.py
View file

@ -1021,7 +1021,7 @@ def _unsloth_get_batch_samples(self, epoch_iterator, num_batches):
if torch.is_tensor(num_items_in_batch):
num_items_in_batch = num_items_in_batch.item()
print("NUM_ITMES = ", num_items_in_batch, type(batch_samples))
print("NUM_ITMES = ", num_items_in_batch, type(batch_samples), self.model)
return batch_samples, num_items_in_batch