fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-06 17:50:25 +08:00
Code Activity

修复了Onnx Attention的一个Shape错误 (#2003)

Browse Source 
* Add files via upload

* Fix Onnx Bugs

* Add files via upload
This commit is contained in:
Ναρουσέ·μ·γιουμεμί·Χινακάννα
2024-04-23 02:07:13 +08:00
committed by GitHub
parent fb69a36068
commit deca566ab2
4 changed files with 476 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
infer/modules/onnx/export.py
View File

@@ -25,7 +25,7 @@ def export_onnx(ModelPath, ExportedPath):
output_names = [
"audio",
]
# net_g.construct_spkmixmap(n_speaker) 多角色混合轨道导出
#net_g.construct_spkmixmap() #多角色混合轨道导出
torch.onnx.export(
net_g,
(
@@ -44,8 +44,8 @@ def export_onnx(ModelPath, ExportedPath):
"rnd": [2],
},
do_constant_folding=False,
opset_version=18,
verbose=True,
opset_version=17,
verbose=False,
input_names=input_names,
output_names=output_names,
)