Update modeling_whisper.py
Browse files- 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.
|
| 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
|