Update modeling_whisper.py
Browse files- 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)
|