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

chore(format): run black on dev (#2)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2024-06-05 21:30:15 +09:00
committed by GitHub
parent 6ff713c024
commit e2fcbff1e9
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rvc/onnx/infer.py
View File

@@ -6,6 +6,7 @@ from onnx.f0predictor import PMF0Predictor
from onnx.f0predictor import HarvestF0Predictor from onnx.f0predictor import HarvestF0Predictor
from onnx.f0predictor import DioF0Predictor from onnx.f0predictor import DioF0Predictor
class ContentVec: class ContentVec:
def __init__(self, vec_path: str, device=None): def __init__(self, vec_path: str, device=None):
if device == "cpu" or device is None: if device == "cpu" or device is None:
@@ -30,12 +31,14 @@ class ContentVec:
logits = self.model.run(None, onnx_input)[0] logits = self.model.run(None, onnx_input)[0]
return logits.transpose(0, 2, 1) return logits.transpose(0, 2, 1)
predicters = { predicters = {
"pm": PMF0Predictor, "pm": PMF0Predictor,
"harvest": HarvestF0Predictor, "harvest": HarvestF0Predictor,
"dio": DioF0Predictor, "dio": DioF0Predictor,
} }
def get_f0_predictor(f0_method, hop_length, sampling_rate): def get_f0_predictor(f0_method, hop_length, sampling_rate):
return predicters[f0_method](hop_length=hop_length, sampling_rate=sampling_rate) return predicters[f0_method](hop_length=hop_length, sampling_rate=sampling_rate)

2
tools/onnx/onnx_inference_demo.py
View File

@@ -9,7 +9,7 @@ f0_up_key = 0 # 升降调
sid = 0 # 角色ID sid = 0 # 角色ID
f0_method = "dio" # F0提取算法 f0_method = "dio" # F0提取算法
model_path = "exported_model.onnx" # 模型的完整路径 model_path = "exported_model.onnx" # 模型的完整路径
vec_path = "vec-256-layer-9.onnx" # 需要onnx的vec模型 vec_path = "vec-256-layer-9.onnx" # 需要onnx的vec模型
wav_path = "123.wav" # 输入路径或ByteIO实例 wav_path = "123.wav" # 输入路径或ByteIO实例
out_path = "out.wav" # 输出路径或ByteIO实例 out_path = "out.wav" # 输出路径或ByteIO实例