chore: remove hard code value to tests

huggingface · Jan 28, 2024 · 0aa3bc9 · 0aa3bc9
1 parent 00905fc
commit 0aa3bc9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/integrations/deepspeed.py b/src/transformers/integrations/deepspeed.py
@@ -458,7 +458,7 @@ def deepspeed_load_checkpoint(
         # this magically updates self.optimizer and self.lr_scheduler
         load_path, _ = deepspeed_engine.load_checkpoint(
             checkpoint_path,
-            load_module_strict=False,
+            load_module_strict=load_module_strict,
             load_optimizer_states=True,
             load_lr_scheduler_states=True,
         )