Apply suggestion from @MahmoudAshraf97

Co-authored-by: Mahmoud Ashraf <hassouna97.ma@gmail.com>
Signed-off-by: Piotr Żelasko <petezor@gmail.com>

Author: pzelasko

scripts/speech_recognition/convert_torchaudio_nemo.py

@@ -63,7 +63,10 @@ def migrate_state_dict(state_dict: dict) -> tuple[dict, list[tuple[str, str]]]:
         for old_suffix, new_suffix in KEY_MIGRATION.items():
             if key.endswith(old_suffix):
                 new_key = key[: -len(old_suffix)] + new_suffix
-                state_dict[new_key] = state_dict.pop(key)
+               if "featurizer.fb" in new_suffix:
+                    state_dict[new_key] = state_dict.pop(key).T.unsqueeze(0)
+                else:
+                    state_dict[new_key] = state_dict.pop(key)
                 renames.append((key, new_key))
                 break
     return state_dict, renames