From 45c858a5d8b41a91f160d4d0901f5efeeb54201b Mon Sep 17 00:00:00 2001 From: Cyril Vallez Date: Thu, 6 Feb 2025 18:38:30 +0100 Subject: [PATCH] fix --- 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 9d893b201..6594849fe 100755 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -1424,7 +1424,7 @@ def _find_missing_and_unexpected_keys( expected_keys = hf_quantizer.update_expected_keys(model, expected_keys, original_loaded_keys) loaded_keys = _adjust_loaded_keys_prefix( - renamed_loaded_keys, loading_base_model_from_task_state_dict, loading_task_model_from_base_state_dict + renamed_loaded_keys, prefix, loading_base_model_from_task_state_dict, loading_task_model_from_base_state_dict ) # Adjust prefix of the keys to make them match loaded keys before removing them