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