1NEYRON1 commited on
Commit
dd68667
·
verified ·
1 Parent(s): 380d35f

Update modeling_whisper.py

Browse files
Files changed (1) hide show
  1. modeling_whisper.py +2 -2
modeling_whisper.py CHANGED
@@ -500,5 +500,5 @@ class WhisperSSLEnsemble(PreTrainedModel): # type: ignore
500
  return scores
501
 
502
 
503
- AutoConfig.register("mosnet", WhisperConfig)
504
- AutoModel.register(WhisperConfig, WhisperSSLEnsemble)
 
500
  return scores
501
 
502
 
503
+ AutoConfig.register("mosnet", WhisperSSLEnsembleConfig)
504
+ AutoModel.register(WhisperSSLEnsembleConfig, WhisperSSLEnsemble)