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

fix

Browse source
This commit is contained in:
Cyril Vallez 2025-02-06 18:38:30 +01:00
parent 3b56ad06c4
commit 45c858a5d8
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

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