From 7ab7dba5c8d71625aa231e8982f0992011ac8712 Mon Sep 17 00:00:00 2001 From: w-okada Date: Sun, 23 Jul 2023 21:19:45 +0900 Subject: [PATCH] bugfix: initialize gpu --- server/voice_changer/RVC/RVC.py | 2 +- server/voice_changer/VoiceChangerManager.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/server/voice_changer/RVC/RVC.py b/server/voice_changer/RVC/RVC.py index c06c25c2..69f48663 100644 --- a/server/voice_changer/RVC/RVC.py +++ b/server/voice_changer/RVC/RVC.py @@ -233,7 +233,7 @@ class RVC(VoiceChangerModel): self.initialize() output_file_simple = export2onnx(self.settings.gpu, modelSlot) - + return { "status": "ok", "path": f"/tmp/{output_file_simple}", diff --git a/server/voice_changer/VoiceChangerManager.py b/server/voice_changer/VoiceChangerManager.py index c4013083..f8de5c1a 100644 --- a/server/voice_changer/VoiceChangerManager.py +++ b/server/voice_changer/VoiceChangerManager.py @@ -81,6 +81,8 @@ class VoiceChangerManager(ServerDeviceCallbacks): self.stored_setting = json.load(open(STORED_SETTING_FILE, "r", encoding="utf-8")) if "modelSlotIndex" in self.stored_setting: 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(): # self.update_settings(key, val)