Spaces:
Runtime error
Runtime error
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
|