Update configuration_whisper.py

configuration_whisper.py

@@ -15,6 +15,7 @@ class WhisperSSLEnsembleConfig(PretrainedConfig):
         weights: str = None,
         device: str | None = None,
         ssl_ensemble_config: Optional[Dict] = None,
+        whisper_embedding_dim: int = 512,
         **kwargs,
     ):
         super().__init__(**kwargs)
@@ -22,4 +23,5 @@ class WhisperSSLEnsembleConfig(PretrainedConfig):
         self.text_model_type = text_model_type
         self.whisper_weights_path = weights
         self.device = device
-        self.ssl_ensemble_config = ssl_ensemble_config
+        self.ssl_ensemble_config = ssl_ensemble_config
+        self.whisper_embedding_dim = whisper_embedding_dim