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

update

Browse Source
This commit is contained in:
wataru 2023-04-14 15:25:52 +09:00
parent bb769cf6c2
commit 4171baa6de
1 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -99,6 +99,7 @@ class SynthesizerTrnMs256NSFsid_ONNX(nn.Module):
def export2onnx(input_model, output_model, output_model_simple, is_half): def export2onnx(input_model, output_model, output_model_simple, is_half):
is_half = False
cpt = torch.load(input_model, map_location="cpu") cpt = torch.load(input_model, map_location="cpu")
if is_half: if is_half:
dev = torch.device("cuda", index=0) dev = torch.device("cuda", index=0)
@ -106,20 +107,10 @@ def export2onnx(input_model, output_model, output_model_simple, is_half):
dev = torch.device("cpu") dev = torch.device("cpu")
net_g_onnx = SynthesizerTrnMs256NSFsid_ONNX(*cpt["config"], is_half=is_half) net_g_onnx = SynthesizerTrnMs256NSFsid_ONNX(*cpt["config"], is_half=is_half)
try: net_g_onnx.eval().to(dev)
net_g_onnx.eval().to(dev)
except:
is_half = False
dev = torch.device("cpu")
net_g_onnx.eval().to(dev)
net_g_onnx.load_state_dict(cpt["weight"], strict=False) net_g_onnx.load_state_dict(cpt["weight"], strict=False)
if is_half: if is_half:
print("!!!!!!!!!!!!!!!!!! half")
net_g_onnx = net_g_onnx.half() net_g_onnx = net_g_onnx.half()
else:
print("!!!!!!!!!!!!!!!!!! full")
net_g_onnx = net_g_onnx.float()
if is_half: if is_half:
feats = torch.HalfTensor(1, 2192, 256).to(dev) feats = torch.HalfTensor(1, 2192, 256).to(dev)