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-05 01:10:22 +08:00
Code Activity

optimize(infer): move onnx into rvc

Browse Source
This commit is contained in:
源文雨
2024-06-11 17:21:05 +09:00
parent e81b7c52c0
commit f956b333fa
12 changed files with 108 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/onnx/export.py
View File

@@ -1,3 +1,3 @@
from infer.modules.onnx.export import export_onnx
from rvc.onnx import export_onnx
export_onnx("pt/Justin Bieber.pth", "pt/TestRvc_Rvc.onnx")

4
tools/onnx/infer.py
View File

@@ -14,11 +14,11 @@ wav_path = "123.wav" # 输入路径或ByteIO实例
out_path = "out.wav" # 输出路径或ByteIO实例
model = RVC(
model_path, vec_path=vec_path, sr=sampling_rate, hop_len=hop_size, device="cuda"
model_path, vec_path=vec_path, hop_len=hop_size, device="cuda"
)
wav, sr = librosa.load(wav_path, sr=sampling_rate)
audio = model.infer(wav, sr, sid, f0_method=f0_method, f0_up_key=f0_up_key)
audio = model.infer(wav, sr, sampling_rate, sid, f0_method, f0_up_key)
soundfile.write(out_path, audio, sampling_rate)