diff --git a/src/transformers/integrations/integration_utils.py b/src/transformers/integrations/integration_utils.py index 1f20ed2e7e1..bcfebdbd113 100755 --- a/src/transformers/integrations/integration_utils.py +++ b/src/transformers/integrations/integration_utils.py @@ -938,6 +938,7 @@ class WandbCallback(TrainerCallback): 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"] )