saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update modeling_utils.py

Browse source
This commit is contained in:
Cyril Vallez 2025-02-05 13:53:54 +01:00
parent 11e378024d
commit a6ffd7b4d2
No known key found for this signature in database
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/transformers/modeling_utils.py
View file

@ -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