Skip to content

Commit 4cab509

Browse files
ljcc0930tjruwase
authored andcommitted
Add default value to "checkpoint_folder" in "load_state_dict" of bf16_optimizer (deepspeedai#6446)
Add default value `checkpoint_folder=None` for compatibility. Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
1 parent 7f2dc25 commit 4cab509

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepspeed/runtime/bf16_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,7 +482,7 @@ def refresh_fp32_params(self):
482482

483483
def load_state_dict(self,
484484
state_dict_list,
485-
checkpoint_folder,
485+
checkpoint_folder=None,
486486
load_optimizer_states=True,
487487
load_from_fp32_weights=False,
488488
load_serial=None,

0 commit comments

Comments
 (0)