aadnk commited on
Commit
9939d13
1 Parent(s): 295de00

Fix convert_hf_whisper

Browse files
Files changed (1) hide show
  1. src/conversion/hf_converter.py +1 -1
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