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

bugfix: export onnx

Browse Source
This commit is contained in:
wataru 2023-04-22 15:12:10 +09:00
parent 8a17da84cf
commit 5cc45dcc2e
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -383,20 +383,23 @@ class RVC:
if hasattr(self, "net_g") == False or self.net_g == None: if hasattr(self, "net_g") == False or self.net_g == None:
print("[Voice Changer] export2onnx, No pyTorch session.") print("[Voice Changer] export2onnx, No pyTorch session.")
return {"status": "ng", "path": f""} return {"status": "ng", "path": f""}
if self.settings.pyTorchModelFile == None:
pyTorchModelFile = self.settings.modelSlots[self.currentSlot].pyTorchModelFile
if pyTorchModelFile == None:
print("[Voice Changer] export2onnx, No pyTorch filepath.") print("[Voice Changer] export2onnx, No pyTorch filepath.")
return {"status": "ng", "path": f""} return {"status": "ng", "path": f""}
import voice_changer.RVC.export2onnx as onnxExporter import voice_changer.RVC.export2onnx as onnxExporter
output_file = os.path.splitext(os.path.basename(self.settings.pyTorchModelFile))[0] + ".onnx" output_file = os.path.splitext(os.path.basename(pyTorchModelFile))[0] + ".onnx"
output_file_simple = os.path.splitext(os.path.basename(self.settings.pyTorchModelFile))[0] + "_simple.onnx" output_file_simple = os.path.splitext(os.path.basename(pyTorchModelFile))[0] + "_simple.onnx"
output_path = os.path.join(TMP_DIR, output_file) output_path = os.path.join(TMP_DIR, output_file)
output_path_simple = os.path.join(TMP_DIR, output_file_simple) output_path_simple = os.path.join(TMP_DIR, output_file_simple)
if torch.cuda.device_count() > 0: if torch.cuda.device_count() > 0:
onnxExporter.export2onnx(self.settings.pyTorchModelFile, output_path, output_path_simple, True) onnxExporter.export2onnx(pyTorchModelFile, output_path, output_path_simple, True)
else: else:
print("[Voice Changer] Warning!!! onnx export with float32. maybe size is doubled.") print("[Voice Changer] Warning!!! onnx export with float32. maybe size is doubled.")
onnxExporter.export2onnx(self.settings.pyTorchModelFile, output_path, output_path_simple, False) onnxExporter.export2onnx(pyTorchModelFile, output_path, output_path_simple, False)
return {"status": "ok", "path": f"/tmp/{output_file_simple}", "filename": output_file_simple} return {"status": "ok", "path": f"/tmp/{output_file_simple}", "filename": output_file_simple}