Update src/transformers/trainer_callback.py

Co-authored-by: Sourab Mangrulkar <[email protected]>

src/transformers/trainer_callback.py

@@ -460,7 +460,7 @@ def on_step_end(self, args: TrainingArguments, state: TrainerState, control: Tra
         # End training
         if state.global_step >= state.max_steps:
             control.should_training_stop = True
-            # Save the best model at the end if we have a save strategy
+            # Save the model at the end if we have a save strategy
             if args.save_strategy != IntervalStrategy.NO:
                 control.should_save = True