fix

huggingface · Dec 15, 2023 · 0668589 · 0668589
1 parent dec51b6
commit 0668589
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
@@ -4163,7 +4163,8 @@ def _find_mismatched_keys(
 
         if _fast_init:
             mismatched_model_keys = [
-                checkpoint_key_to_model_key(x[0], remove_prefix_from_model, add_prefix_to_model) for x in mismatched_keys
+                checkpoint_key_to_model_key(x[0], remove_prefix_from_model, add_prefix_to_model)
+                for x in mismatched_keys
             ]
             set_initialized_submodules(model, mismatched_model_keys, loaded=False)
             # This will only initialize submodules that are re-marked as `not loaded` above due to mismatched