PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-02-03 08:43:57 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix: DDSP SVC initail update

Browse Source
This commit is contained in:
wataru 2023-04-17 10:35:16 +09:00
parent e4ac5e74db
commit 2d19b5ff65
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/voice_changer/DDSP_SVC/DDSP_SVC.py
View File

@ -73,7 +73,7 @@ class DDSP_SVC:
return torch.device("cpu") return torch.device("cpu")
def loadModel(self, props): def loadModel(self, props):
self.settings.configFile = props["files"]["configFilename"] # self.settings.configFile = props["files"]["configFilename"] # 同じフォルダにあるyamlを使う
self.settings.pyTorchModelFile = props["files"]["pyTorchModelFilename"] self.settings.pyTorchModelFile = props["files"]["pyTorchModelFilename"]
# model # model
model, args = vo.load_model(self.settings.pyTorchModelFile, device=self.useDevice()) model, args = vo.load_model(self.settings.pyTorchModelFile, device=self.useDevice())
@ -136,7 +136,7 @@ class DDSP_SVC:
if "CUDAExecutionProvider" in providers: if "CUDAExecutionProvider" in providers:
provider_options = [{'device_id': self.settings.gpu}] provider_options = [{'device_id': self.settings.gpu}]
self.onnx_session.set_providers(providers=["CUDAExecutionProvider"], provider_options=provider_options) self.onnx_session.set_providers(providers=["CUDAExecutionProvider"], provider_options=provider_options)
if key == "gpu": if key == "gpu" and len(self.settings.pyTorchModelFile) > 0:
model, _args = vo.load_model(self.settings.pyTorchModelFile, device=self.useDevice()) model, _args = vo.load_model(self.settings.pyTorchModelFile, device=self.useDevice())
self.model = model self.model = model
self.enhancer = Enhancer(self.args.enhancer.type, self.enhancer_path, device=self.useDevice()) self.enhancer = Enhancer(self.args.enhancer.type, self.enhancer_path, device=self.useDevice())