🐛[bugfix] Config: cpu/cudaの保存するdictの形式を修正

This commit is contained in:
misyaguziya
2024-10-23 14:13:54 +09:00
parent af3fe1f0f9
commit 5b44ddefde
2 changed files with 5 additions and 5 deletions

View File

@@ -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]"