From 367be9019bb266e286f4ce52428903b283372be6 Mon Sep 17 00:00:00 2001 From: Cyril Vallez Date: Tue, 4 Feb 2025 21:15:34 +0100 Subject: [PATCH] Update modeling_utils.py --- src/transformers/modeling_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 658f9c911..59b9f690f 100755 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -4720,7 +4720,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix loaded_keys = [cls._fix_state_dict_key_on_load(key)[0] for key in loaded_state_dict_keys] has_prefix_module = any(s.startswith(prefix) for s in loaded_keys) if len(prefix) > 0 else False - expects_prefix_module = any(s.startswith(prefix) for s in expected_keys) if len(len(prefix) > 0) else False + expects_prefix_module = any(s.startswith(prefix) for s in expected_keys) if len(prefix) > 0 else False remove_prefix_from_model = not has_prefix_module and expects_prefix_module add_prefix_to_model = has_prefix_module and not expects_prefix_module