Skip to content

Commit ddd3571

Browse files
ljcc0930tjruwase
andauthored
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]>
1 parent 5d1a30c commit ddd3571

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: deepspeed/runtime/bf16_optimizer.py

+1-1
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)