SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix convert to original state dict for VLMs (#38385)

Browse Source 
* fix convert to original state dict

* fix

* lint

* Update modeling_utils.py
This commit is contained in:
hoshi-hiyouga
2025-05-27 18:27:59 +08:00
committed by GitHub
parent c769483188
commit 008e0d87c5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_utils.py
View File

@@ -3653,7 +3653,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, PushToHubMixin, PeftAdapterMi
for key, value in state_dict.items(): for key, value in state_dict.items():
for pattern, replacement in reverse_key_mapping.items(): for pattern, replacement in reverse_key_mapping.items():
replacement = replacement.lstrip("^") # strip off un-needed chars and patterns replacement = replacement.lstrip("^") # strip off un-needed chars and patterns
replacement = re.sub(r"\(.*?\)", "", pattern) replacement = re.sub(r"\(.*\)", "", replacement)
key, n_replace = re.subn(pattern, replacement, key) key, n_replace = re.subn(pattern, replacement, key)
# Early exit of the loop # Early exit of the loop
if n_replace > 0: if n_replace > 0: