every_holiday/VRCT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Update] Rename and add methods for CTranslate2 compute type management in Controller

Browse Source
This commit is contained in:
misyaguziya
2025-09-17 10:52:56 +09:00
parent 245855d0ca
commit 5f0f918642
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src-python/controller.py
View File

@@ -1456,11 +1456,15 @@ class Controller:
return {"status":200, "result":config.CTRANSLATE2_WEIGHT_TYPE}
@staticmethod
def getCtranslateComputeType(*args, **kwargs) -> dict:
def getCtranslate2ComputeTypeList(*args, **kwargs) -> dict:
return {"status":200, "result":config.SELECTABLE_CTRANSLATE2_COMPUTE_TYPE_LIST}
@staticmethod
def getCtranslate2ComputeType(*args, **kwargs) -> dict:
return {"status":200, "result":config.CTRANSLATE2_COMPUTE_TYPE}
@staticmethod
def setCtranslateComputeType(data, *args, **kwargs) -> dict:
def setCtranslate2ComputeType(data, *args, **kwargs) -> dict:
config.CTRANSLATE2_COMPUTE_TYPE = str(data)
if model.checkTranslatorCTranslate2ModelWeight(config.CTRANSLATE2_WEIGHT_TYPE):
def callback():