1NEYRON1 commited on
Commit
df60c0d
·
verified ·
1 Parent(s): 488d67c

Update modeling_whisper.py

Browse files
Files changed (1) hide show
  1. modeling_whisper.py +1 -1
modeling_whisper.py CHANGED
@@ -864,7 +864,7 @@ class WhisperSSLEnsemble(PreTrainedModel): # type: ignore
864
  # to the target device. This is the correct way to set the device for the model.
865
  # This also correctly sets the self.device property.
866
  self.to(self._target_device)
867
- logger.info(f"WhisperSSLEnsemble initialization complete. Final model device: {self.device}")
868
 
869
  # Поместите этот метод внутри класса WhisperSSLEnsemble
870
  # В классе WhisperSSLEnsemble
 
864
  # to the target device. This is the correct way to set the device for the model.
865
  # This also correctly sets the self.device property.
866
  self.to(self._target_device)
867
+ logger.info(f"WhisperSSLEnsemble initialization complete. Final model device: {self._target_device}")
868
 
869
  # Поместите этот метод внутри класса WhisperSSLEnsemble
870
  # В классе WhisperSSLEnsemble