1NEYRON1 commited on
Commit
440e6ce
·
verified ·
1 Parent(s): c84e9a4

Update configuration_whisper.py

Browse files
Files changed (1) hide show
  1. 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 = 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
  ):