update rmvpe checkpoint

This commit is contained in:
w-okada 2023-10-09 14:24:56 +09:00
parent 70ba5d9f24
commit 9b39111f46
2 changed files with 3 additions and 3 deletions

View File

@ -105,7 +105,7 @@ def downloadWeight(voiceChangerParams: VoiceChangerParams):
if os.path.exists(rmvpe) is False:
downloadParams.append(
{
"url": "https://huggingface.co/wok000/weights/resolve/main/rmvpe/rmvpe.pt",
"url": "https://huggingface.co/wok000/weights/resolve/main/rmvpe/rmvpe_20231006.pt",
"saveTo": rmvpe,
"position": 8,
}
@ -113,7 +113,7 @@ def downloadWeight(voiceChangerParams: VoiceChangerParams):
if os.path.exists(rmvpe_onnx) is False:
downloadParams.append(
{
"url": "https://huggingface.co/wok000/weights_gpl/resolve/main/rmvpe/rmvpe_s.onnx",
"url": "https://huggingface.co/wok000/weights_gpl/resolve/main/rmvpe/rmvpe_20231006.onnx",
"saveTo": rmvpe_onnx,
"position": 9,
}

View File

@ -52,7 +52,7 @@ class RMVPEOnnxPitchExtractor(PitchExtractor):
audio = np.expand_dims(audio, axis=0)
output = self.onnx_session.run(
["f0", "uv"],
["pitchf"],
{
"waveform": audio.astype(np.float32),
"threshold": np.array([0.3]).astype(np.float32),