From 59ffd06268137ca6cbc042e035af6cc071b0c3a7 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 26 Dec 2024 23:37:25 -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 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):