From 52b24512de064080096ec7949fbe48efbeef8aca Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 26 Dec 2024 22:25:19 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 18918a3c76..986b938f1b 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -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