1NEYRON1 commited on
Commit
4d58aec
·
verified ·
1 Parent(s): 09a02b7

Update modeling_whisper.py

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