diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 30adf05ac6e8..8923b9d5c9ac 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -3653,7 +3653,7 @@ def save_pretrained( for key, value in state_dict.items(): for pattern, replacement in reverse_key_mapping.items(): 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) # Early exit of the loop if n_replace > 0: