From a6ffd7b4d2f214b70ee6c6b2dd85ecf7a3633bb8 Mon Sep 17 00:00:00 2001 From: Cyril Vallez Date: Wed, 5 Feb 2025 13:53:54 +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 446f8847f..e3bf09f02 100755 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -4757,7 +4757,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix # they are not in the loaded state dict) if low_cpu_mem_usage: model = move_missing_keys_back_to_cpu( - model, missing_keys, unexpected_keys, dtype, keep_in_fp32_modules, is_quantized, hf_quantizer + model, missing_keys, unexpected_keys, dtype, keep_in_fp32_modules, hf_quantizer ) # correctly initialize the missing keys