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

Merge branch 'cuda' into for_webui

Browse Source
This commit is contained in:
misyaguziya
2024-12-02 10:06:01 +09:00
parent 6017f013a1 9d62906903
commit e25abe11eb
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src-python/webui_controller.py
View File

@@ -407,6 +407,14 @@ class Controller:
def getMessageBoxRatioRange(*args, **kwargs) -> dict: def getMessageBoxRatioRange(*args, **kwargs) -> dict:
return {"status":200, "result":config.MESSAGE_BOX_RATIO_RANGE} return {"status":200, "result":config.MESSAGE_BOX_RATIO_RANGE}
@staticmethod
def getComputeMode(*args, **kwargs) -> dict:
if torch.cuda.is_available():
compute_mode = "cuda"
else:
compute_mode = "cpu"
return {"status":200, "result":compute_mode}
@staticmethod @staticmethod
def getComputeDeviceList(*args, **kwargs) -> dict: def getComputeDeviceList(*args, **kwargs) -> dict:
device_list = [] device_list = []

1
src-python/webui_mainloop.py
View File

@@ -133,6 +133,7 @@ mapping = {
"/set/data/main_window_geometry": {"status": True, "variable":controller.setMainWindowGeometry}, "/set/data/main_window_geometry": {"status": True, "variable":controller.setMainWindowGeometry},
# Translation # Translation
"/get/data/compute_mode": {"status": True, "variable":controller.getComputeMode},
"/get/data/translation_compute_device_list": {"status": True, "variable":controller.getComputeDeviceList}, "/get/data/translation_compute_device_list": {"status": True, "variable":controller.getComputeDeviceList},
"/get/data/selected_translation_compute_device": {"status": True, "variable":controller.getSelectedTranslationComputeDevice}, "/get/data/selected_translation_compute_device": {"status": True, "variable":controller.getSelectedTranslationComputeDevice},
"/set/data/selected_translation_compute_device": {"status": True, "variable":controller.setSelectedTranslationComputeDevice}, "/set/data/selected_translation_compute_device": {"status": True, "variable":controller.setSelectedTranslationComputeDevice},