"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "e27d9308be940c8372aec7187553294628c30043"
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,7 +1050,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
# Handle the case where some state_dict keys shouldn't be saved
if self._keys_to_ignore_on_save is not None:
for ignore_key in self._keys_to_ignore_on_save:
del state_dict[ignore_key]
if ignore_key in state_dict.keys():
del state_dict[ignore_key]
# If we save using the predefined names, we can load using `from_pretrained`
output_model_file = os.path.join(save_directory, WEIGHTS_NAME)
......
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