diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index c1bc7aa972..9725f624ad 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1025,9 +1025,9 @@ def _unsloth_get_batch_samples(self, epoch_iterator, num_batches): num_items_in_batch = num_items_in_batch.item() except Exception as exception: logger.warning_once(exception) - pass pass + print(batch_samples, num_items_in_batch) return batch_samples, num_items_in_batch # def _unsloth_get_batch_samples(self, epoch_iterator, num_batches):