fix: prevent second save in the end of training

Author: JaktensTid

src/transformers/trainer_callback.py

@@ -604,7 +604,7 @@ def on_step_end(self, args: TrainingArguments, state: TrainerState, control: Tra
         if state.global_step >= state.max_steps:
             control.should_training_stop = True
             # Save the model at the end if we have a save strategy
-            if args.save_strategy not in [SaveStrategy.NO, SaveStrategy.BEST, SaveStrategy.STEPS]:
+            if args.save_strategy == SaveStrategy.STEPS:
                 control.should_save = True
 
         return control