Fix: Unexpected Keys, Improve run_compressed, Rename Test Folder (#37077)

2025-04-04 14:30:11 -05:00
parent 531e4fcf0e
commit ebe47ce3e9
6 changed files with 48 additions and 32 deletions
--- a/src/transformers/modeling_utils.py
+++ b/src/transformers/modeling_utils.py
@@ -1352,6 +1352,7 @@ def _find_missing_and_unexpected_keys(

    if hf_quantizer is not None:
        missing_keys = hf_quantizer.update_missing_keys(model, missing_keys, prefix)
+        unexpected_keys = hf_quantizer.update_unexpected_keys(model, unexpected_keys, prefix)

    # Model-specific exceptions for missing and unexpected keys (e.g. if the modeling change over time, or any other reason...)
    if cls._keys_to_ignore_on_load_missing is not None: