Update modeling_whisper.py
Browse files- modeling_whisper.py +4 -0
modeling_whisper.py
CHANGED
|
@@ -272,6 +272,10 @@ class WhisperSSLEnsemble(PreTrainedModel): # type: ignore
|
|
| 272 |
|
| 273 |
super().__init__(config)
|
| 274 |
self.config = config
|
|
|
|
|
|
|
|
|
|
|
|
|
| 275 |
# device = 'cpu'
|
| 276 |
# if self.config.device is None:
|
| 277 |
# device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 272 |
|
| 273 |
super().__init__(config)
|
| 274 |
self.config = config
|
| 275 |
+
weights = self.config.weights
|
| 276 |
+
whisper_variant = self.config.whisper_variant
|
| 277 |
+
text_model_type = self.config.text_model_type
|
| 278 |
+
ssl_ensemble_config = self.config.ssl_ensemble_config
|
| 279 |
# device = 'cpu'
|
| 280 |
# if self.config.device is None:
|
| 281 |
# device = "cuda" if torch.cuda.is_available() else "cpu"
|