From 5b44ddefde6a319625e08213a9e896a1d4286323 Mon Sep 17 00:00:00 2001
From: misyaguziya <53165965+misyaguziya@users.noreply.github.com>
Date: Wed, 23 Oct 2024 14:13:54 +0900
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B[bugfix]=20Config:=20cpu/cuda?=
 =?UTF-8?q?=E3=81=AE=E4=BF=9D=E5=AD=98=E3=81=99=E3=82=8Bdict=E3=81=AE?=
 =?UTF-8?q?=E5=BD=A2=E5=BC=8F=E3=82=92=E4=BF=AE=E6=AD=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src-python/config.py | 4 ++--
 src-python/model.py  | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

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():