null deepspeed_plugin in args for wandb callback fake trainer (#38867)

winglian · web-flow · commit 12d4c5b66f7f · 2025-06-18T13:10:22.000Z
diff --git a/src/transformers/integrations/integration_utils.py b/src/transformers/integrations/integration_utils.py
@@ -938,6 +938,7 @@ def on_train_end(self, args: TrainingArguments, state, control, model=None, proc
 
             args_for_fake = copy.deepcopy(args)
             args_for_fake.deepspeed = None
+            args_for_fake.deepspeed_plugin = None
             fake_trainer = Trainer(
                 args=args_for_fake, model=model, processing_class=processing_class, eval_dataset=["fake"]
             )

Original file line number	Diff line number	Diff line change
`@@ -938,6 +938,7 @@ def on_train_end(self, args: TrainingArguments, state, control, model=None, proc`
`938`	`938`
`939`	`939`	`args_for_fake = copy.deepcopy(args)`
`940`	`940`	`args_for_fake.deepspeed = None`
	`941`	`+ args_for_fake.deepspeed_plugin = None`
`941`	`942`	`fake_trainer = Trainer(`
`942`	`943`	`args=args_for_fake, model=model, processing_class=processing_class, eval_dataset=["fake"]`
`943`	`944`	`)`