mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-02-02 16:23:58 +03:00
update rmvpe checkpoint
This commit is contained in:
parent
70ba5d9f24
commit
9b39111f46
@ -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,
|
||||
}
|
||||
|
@ -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),
|
||||
|
Loading…
Reference in New Issue
Block a user