Skip to content

Commit 2842b82

Browse files
hiyougaArthurZucker
authored andcommitted
Fix convert to original state dict for VLMs (#38385)
* fix convert to original state dict * fix * lint * Update modeling_utils.py
1 parent 24c6d5b commit 2842b82

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/modeling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3533,7 +3533,7 @@ def save_pretrained(
35333533
for key, value in state_dict.items():
35343534
for pattern, replacement in reverse_key_mapping.items():
35353535
replacement = replacement.lstrip("^") # strip off un-needed chars and patterns
3536-
replacement = re.sub(r"\(.*?\)", "", pattern)
3536+
replacement = re.sub(r"\(.*\)", "", replacement)
35373537
key, n_replace = re.subn(pattern, replacement, key)
35383538
# Early exit of the loop
35393539
if n_replace > 0:

0 commit comments

Comments
 (0)