Update configuration_whisper.py
Browse files- configuration_whisper.py +1 -1
configuration_whisper.py
CHANGED
|
@@ -13,7 +13,7 @@ class WhisperSSLEnsembleConfig(PretrainedConfig):
|
|
| 13 |
whisper_variant: str = "base.en",
|
| 14 |
text_model_type: str = "bert-base-uncased",
|
| 15 |
weights: str = None,
|
| 16 |
-
device: str | None =
|
| 17 |
ssl_ensemble_config: Optional[Dict] = None,
|
| 18 |
**kwargs,
|
| 19 |
):
|
|
|
|
| 13 |
whisper_variant: str = "base.en",
|
| 14 |
text_model_type: str = "bert-base-uncased",
|
| 15 |
weights: str = None,
|
| 16 |
+
device: str | None = "cpu",
|
| 17 |
ssl_ensemble_config: Optional[Dict] = None,
|
| 18 |
**kwargs,
|
| 19 |
):
|