Skip to content

GitLab

Unverified Commit 69f2d538 authored by hanrui1sensetime's avatar hanrui1sensetime Committed by GitHub
Browse files

[Fix Bugs] Fix keys in `_load_pretrained_model` (#22947) 

fix transformers keys
parent b5f06d6c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/modeling_utils.py
View file @ 69f2d538
...@@ -2956,8 +2956,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -2956,8 +2956,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
for key in missing_keys: for key in missing_keys:
if key in list(model_state_dict.keys()): if key in list(model_state_dict.keys()):
key = key key = key
elif f"{prefix}.key" in list(model_state_dict.keys()): elif f"{prefix}.{key}" in list(model_state_dict.keys()):
key = f"{prefix}.key" key = f"{prefix}.{key}"
elif key.startswith(prefix) and ".".join(key.split(".")[1:]) in list(model_state_dict.keys()): elif key.startswith(prefix) and ".".join(key.split(".")[1:]) in list(model_state_dict.keys()):
key = ".".join(key.split(".")[1:]) key = ".".join(key.split(".")[1:])
param = model_state_dict[key] param = model_state_dict[key]
......
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