PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-03-12 10:20:11 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix session none bug

Browse Source
This commit is contained in:
wataru 2023-01-09 21:52:39 +09:00
parent 4d3a725d5d
commit e6d635b2ec
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server/voice_changer/VoiceChanger.py
View File

@ -68,6 +68,8 @@ class VoiceChanger():
**self.hps.model) **self.hps.model)
self.net_g.eval() self.net_g.eval()
utils.load_checkpoint(pyTorch_model_file, self.net_g, None) utils.load_checkpoint(pyTorch_model_file, self.net_g, None)
if hasattr(self, "net_g") == False:
self.net_g = None
# ONNXモデル生成 # ONNXモデル生成
if onnx_model_file != None: if onnx_model_file != None:
@ -77,6 +79,8 @@ class VoiceChanger():
onnx_model_file, onnx_model_file,
providers=providers providers=providers
) )
if hasattr(self, "onnx_session") == False:
self.onnx_session = None
def destroy(self): def destroy(self):
if hasattr(self, "net_g"): if hasattr(self, "net_g"):
@ -86,6 +90,7 @@ class VoiceChanger():
def get_info(self): def get_info(self):
data = asdict(self.settings) data = asdict(self.settings)
data["providers"] = self.onnx_session.get_providers() if self.onnx_session != None else "" data["providers"] = self.onnx_session.get_providers() if self.onnx_session != None else ""
files = ["config_file", "pyTorch_model_file", "onnx_model_file"] files = ["config_file", "pyTorch_model_file", "onnx_model_file"]
for f in files: for f in files: