Fix convert_hf_whisper
Browse files
src/conversion/hf_converter.py
CHANGED
@@ -42,7 +42,7 @@ def rename_keys(s_dict):
|
|
42 |
return s_dict
|
43 |
|
44 |
|
45 |
-
def convert_hf_whisper(hf_model_name_or_path: str, whisper_state_path: str)
|
46 |
from transformers import WhisperForConditionalGeneration
|
47 |
transformer_model = WhisperForConditionalGeneration.from_pretrained(hf_model_name_or_path)
|
48 |
config = transformer_model.config
|
|
|
42 |
return s_dict
|
43 |
|
44 |
|
45 |
+
def convert_hf_whisper(hf_model_name_or_path: str, whisper_state_path: str):
|
46 |
from transformers import WhisperForConditionalGeneration
|
47 |
transformer_model = WhisperForConditionalGeneration.from_pretrained(hf_model_name_or_path)
|
48 |
config = transformer_model.config
|