1NEYRON1 commited on
Commit
8d11de1
·
verified ·
1 Parent(s): 7d069ce

Update configuration_whisper.py

Browse files
Files changed (1) hide show
  1. 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": 'best_ssl_ensemble_model_weak.joblib',
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,