Unverified Commit bec02ff2 authored by Chang Wang's avatar Chang Wang Committed by GitHub
Browse files

enhance rewrite state_dict missing _metadata (#14348)

parent 2b0d9389
...@@ -1050,6 +1050,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -1050,6 +1050,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
# Handle the case where some state_dict keys shouldn't be saved # Handle the case where some state_dict keys shouldn't be saved
if self._keys_to_ignore_on_save is not None: if self._keys_to_ignore_on_save is not None:
for ignore_key in self._keys_to_ignore_on_save: for ignore_key in self._keys_to_ignore_on_save:
if ignore_key in state_dict.keys():
del state_dict[ignore_key] del state_dict[ignore_key]
# If we save using the predefined names, we can load using `from_pretrained` # If we save using the predefined names, we can load using `from_pretrained`
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment