Should make tests pass

huggingface · Jan 8, 2025 · 9a4bbbd · 9a4bbbd
1 parent d9f8f46
commit 9a4bbbd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -2216,6 +2216,7 @@ def _inner_training_loop(
             if args.max_steps < 1:
                 max_steps = math.ceil(args.num_train_epochs * num_update_steps_per_epoch)
 
+        num_train_tokens = None
         if self.args.include_tokens_per_second:
             num_train_tokens = self.num_tokens(train_dataloader, max_steps)
             if has_length(train_dataloader) and args.max_steps < 1: