Add default value to "checkpoint_folder" in "load_state_dict" of bf16…

…_optimizer (#6446) Add default value `checkpoint_folder=None` for compatibility. Co-authored-by: Olatunji Ruwase <[email protected]>
deepspeedai · Sep 4, 2024 · ddd3571 · ddd3571
1 parent 5d1a30c
commit ddd3571
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/runtime/bf16_optimizer.py b/deepspeed/runtime/bf16_optimizer.py
@@ -482,7 +482,7 @@ def refresh_fp32_params(self):
 
     def load_state_dict(self,
                         state_dict_list,
-                        checkpoint_folder,
+                        checkpoint_folder=None,
                         load_optimizer_states=True,
                         load_from_fp32_weights=False,
                         load_serial=None,