Update modeling_whisper.py
Browse files- 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",
|
| 504 |
-
AutoModel.register(
|
|
|
|
| 500 |
return scores
|
| 501 |
|
| 502 |
|
| 503 |
+
AutoConfig.register("mosnet", WhisperSSLEnsembleConfig)
|
| 504 |
+
AutoModel.register(WhisperSSLEnsembleConfig, WhisperSSLEnsemble)
|