1berry/wokada-voice-changer
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bugfix: initialize gpu

Browse Source
This commit is contained in:
w-okada 2023-07-23 21:19:45 +09:00
parent f98d4a38e0
commit 7ab7dba5c8
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/voice_changer/RVC/RVC.py
View File

@ -233,7 +233,7 @@ class RVC(VoiceChangerModel):
self.initialize() self.initialize()
output_file_simple = export2onnx(self.settings.gpu, modelSlot) output_file_simple = export2onnx(self.settings.gpu, modelSlot)
return { return {
"status": "ok", "status": "ok",
"path": f"/tmp/{output_file_simple}", "path": f"/tmp/{output_file_simple}",

2
server/voice_changer/VoiceChangerManager.py
View File

@ -81,6 +81,8 @@ class VoiceChangerManager(ServerDeviceCallbacks):
self.stored_setting = json.load(open(STORED_SETTING_FILE, "r", encoding="utf-8")) self.stored_setting = json.load(open(STORED_SETTING_FILE, "r", encoding="utf-8"))
if "modelSlotIndex" in self.stored_setting: if "modelSlotIndex" in self.stored_setting:
self.update_settings("modelSlotIndex", self.stored_setting["modelSlotIndex"]) self.update_settings("modelSlotIndex", self.stored_setting["modelSlotIndex"])
if "gpu" not in self.stored_setting:
self.update_settings("gpu", 0)
# for key, val in self.stored_setting.items(): # for key, val in self.stored_setting.items():
# self.update_settings(key, val) # self.update_settings(key, val)