1NEYRON1 commited on
Commit
3de85ab
·
verified ·
1 Parent(s): a87160b

Update modeling_whisper.py

Browse files
Files changed (1) hide show
  1. modeling_whisper.py +1 -1
modeling_whisper.py CHANGED
@@ -507,5 +507,5 @@ class WhisperSSLEnsemble(PreTrainedModel): # type: ignore
507
  return scores
508
 
509
 
510
- AutoConfig.register("whisper", WhisperSSLEnsembleConfig)
511
  AutoModel.register(WhisperSSLEnsembleConfig, WhisperSSLEnsemble)
 
507
  return scores
508
 
509
 
510
+ AutoConfig.register("whisper-bert", WhisperSSLEnsembleConfig)
511
  AutoModel.register(WhisperSSLEnsembleConfig, WhisperSSLEnsemble)