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),