Skip to content

Commit d3be336

Browse files
authored
Update check_completion.py
1 parent 06fa5d2 commit d3be336

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/trainer/unified_checkpoint/check_completion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def check_dynamic_load(args, weight_map, existed_files, is_master_weights=False,
186186
state_dict = get_expected_state_dict(model)
187187

188188
for key in state_dict.keys():
189-
if model._keys_to_ignore_on_load_massing is not None and key in model._keys_to_ignore_on_load_missing:
189+
if model._keys_to_ignore_on_load_missing is not None and key in model._keys_to_ignore_on_load_missing:
190190
continue
191191
if sharding_group.nranks > 1:
192192
static_name = struct2static_name_mappings.get(key, None)

0 commit comments

Comments
 (0)