diff --git a/src-python/config.py b/src-python/config.py index 38459a95..a398cddb 100644 --- a/src-python/config.py +++ b/src-python/config.py @@ -1128,8 +1128,8 @@ class Config: self._USE_EXCLUDE_WORDS = True self._USE_TRANSLATION_FEATURE = True self._USE_WHISPER_FEATURE = False - self._SELECTED_TRANSLATION_COMPUTE_DEVICE = {"type": "cpu", "index": 0, "name":"cpu"} - self._SELECTED_TRANSCRIPTION_COMPUTE_DEVICE = {"type": "cpu", "index": 0, "name":"cpu"} + self._SELECTED_TRANSLATION_COMPUTE_DEVICE = {"device": "cpu", "device_index": 0, "device_name":"cpu"} + self._SELECTED_TRANSCRIPTION_COMPUTE_DEVICE = {"device": "cpu", "device_index": 0, "device_name":"cpu"} self._CTRANSLATE2_WEIGHT_TYPE = "Small" self._WHISPER_WEIGHT_TYPE = "base" self._SEND_MESSAGE_FORMAT = "[message]" diff --git a/src-python/model.py b/src-python/model.py index 0d659d18..8f4a68a9 100644 --- a/src-python/model.py +++ b/src-python/model.py @@ -112,7 +112,7 @@ class Model: self.translator.changeCTranslate2Model( config.PATH_LOCAL, config.CTRANSLATE2_WEIGHT_TYPE, - config.SELECTED_TRANSLATION_COMPUTE_DEVICE["type"], + config.SELECTED_TRANSLATION_COMPUTE_DEVICE["device"], config.SELECTED_TRANSLATION_COMPUTE_DEVICE["device_index"]) def downloadCTranslate2ModelWeight(self, callbackFunc=None): @@ -429,7 +429,7 @@ class Model: transcription_engine=config.SELECTED_TRANSCRIPTION_ENGINE, root=config.PATH_LOCAL, whisper_weight_type=config.WHISPER_WEIGHT_TYPE, - device=config.SELECTED_TRANSCRIPTION_COMPUTE_DEVICE["type"], + device=config.SELECTED_TRANSCRIPTION_COMPUTE_DEVICE["device"], device_index=config.SELECTED_TRANSCRIPTION_COMPUTE_DEVICE["device_index"], ) def sendMicTranscript(): @@ -593,7 +593,7 @@ class Model: transcription_engine=config.SELECTED_TRANSCRIPTION_ENGINE, root=config.PATH_LOCAL, whisper_weight_type=config.WHISPER_WEIGHT_TYPE, - device=config.SELECTED_TRANSCRIPTION_COMPUTE_DEVICE["type"], + device=config.SELECTED_TRANSCRIPTION_COMPUTE_DEVICE["device"], device_index=config.SELECTED_TRANSCRIPTION_COMPUTE_DEVICE["device_index"], ) def sendSpeakerTranscript():