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

Improvement:

Browse Source 
onnx on/off
This commit is contained in:
w-okada 2023-07-08 03:45:02 +09:00
parent caa8c99f14
commit eba5da7970
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/voice_changer/RVC/embedder/EmbedderManager.py
View File

@ -41,6 +41,8 @@ class EmbedderManager:
) -> Embedder: ) -> Embedder:
if embederType == "hubert_base": if embederType == "hubert_base":
try: try:
if cls.params.content_vec_500_onnx_on is False:
raise Exception("[Voice Changer][Embedder] onnx is off")
file = cls.params.content_vec_500_onnx file = cls.params.content_vec_500_onnx
return OnnxContentvec().loadModel(file, dev) return OnnxContentvec().loadModel(file, dev)
except Exception as e: except Exception as e:
@ -52,6 +54,8 @@ class EmbedderManager:
return FairseqHubertJp().loadModel(file, dev, isHalf) return FairseqHubertJp().loadModel(file, dev, isHalf)
elif embederType == "contentvec": elif embederType == "contentvec":
try: try:
if cls.params.content_vec_500_onnx_on is False:
raise Exception("[Voice Changer][Embedder] onnx is off")
file = cls.params.content_vec_500_onnx file = cls.params.content_vec_500_onnx
return OnnxContentvec().loadModel(file, dev) return OnnxContentvec().loadModel(file, dev)
except Exception as e: except Exception as e: