nit

huggingface · Oct 3, 2022 · a6361d4 · a6361d4
1 parent e6ee5b5
commit a6361d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/transformers/models/whisper/convert_openai_whisper_to_tfms.py b/src/transformers/models/whisper/convert_openai_whisper_to_tfms.py
@@ -61,9 +61,9 @@ def rename_keys(s_dict):
     keys = list(s_dict.keys())
     for key in keys:
         new_key = key
-        for k, v in WHISPER_MAPPING.items():
-            if k in key:
-                new_key = new_key.replace(k, v)
+        for old_key, new_key in WHISPER_MAPPING.items():
+            if old_key in key:
+                new_key = new_key.replace(old_key, new_key)
 
         print(f"{key} -> {new_key}")