From 12d4c5b66f7ff2765bacfeee214e7504671824bf Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 18 Jun 2025 09:10:22 -0400 Subject: [PATCH] null deepspeed_plugin in args for wandb callback fake trainer (#38867) --- src/transformers/integrations/integration_utils.py | 1 + 1 file changed, 1 insertion(+) 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"] )