Update configuration_whisper.py
Browse files- configuration_whisper.py +1 -1
configuration_whisper.py
CHANGED
|
@@ -14,7 +14,7 @@ class WhisperSSLEnsembleConfig(PretrainedConfig):
|
|
| 14 |
text_model_type: str = "bert-base-uncased",
|
| 15 |
weights: str = None,
|
| 16 |
device: str | None = None,
|
| 17 |
-
ssl_ensemble_config: Optional[Dict] = {"weak_learners_path": '
|
| 18 |
"meta_learner_path": 'pytorch_model.bin',
|
| 19 |
"audio_dim": 512,
|
| 20 |
"text_dim": 768,
|
|
|
|
| 14 |
text_model_type: str = "bert-base-uncased",
|
| 15 |
weights: str = None,
|
| 16 |
device: str | None = None,
|
| 17 |
+
ssl_ensemble_config: Optional[Dict] = {"weak_learners_path": 'weak.joblib',
|
| 18 |
"meta_learner_path": 'pytorch_model.bin',
|
| 19 |
"audio_dim": 512,
|
| 20 |
"text_dim": 768,
|