From 9b39111f4673c23f8a431a00323f0e59988f8ed1 Mon Sep 17 00:00:00 2001 From: w-okada Date: Mon, 9 Oct 2023 14:24:56 +0900 Subject: [PATCH] update rmvpe checkpoint --- server/downloader/WeightDownloader.py | 4 ++-- .../RVC/pitchExtractor/RMVPEOnnxPitchExtractor.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/server/downloader/WeightDownloader.py b/server/downloader/WeightDownloader.py index 6c2427cb..20914ba8 100644 --- a/server/downloader/WeightDownloader.py +++ b/server/downloader/WeightDownloader.py @@ -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, } diff --git a/server/voice_changer/RVC/pitchExtractor/RMVPEOnnxPitchExtractor.py b/server/voice_changer/RVC/pitchExtractor/RMVPEOnnxPitchExtractor.py index b87253b3..09820654 100644 --- a/server/voice_changer/RVC/pitchExtractor/RMVPEOnnxPitchExtractor.py +++ b/server/voice_changer/RVC/pitchExtractor/RMVPEOnnxPitchExtractor.py @@ -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),