PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-02-09 03:37:51 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

update rmvpe checkpoint

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/downloader/WeightDownloader.py
View File

@ -105,7 +105,7 @@ def downloadWeight(voiceChangerParams: VoiceChangerParams):
if os.path.exists(rmvpe) is False: if os.path.exists(rmvpe) is False:
downloadParams.append( 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, "saveTo": rmvpe,
"position": 8, "position": 8,
} }
@ -113,7 +113,7 @@ def downloadWeight(voiceChangerParams: VoiceChangerParams):
if os.path.exists(rmvpe_onnx) is False: if os.path.exists(rmvpe_onnx) is False:
downloadParams.append( 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, "saveTo": rmvpe_onnx,
"position": 9, "position": 9,
} }

2
server/voice_changer/RVC/pitchExtractor/RMVPEOnnxPitchExtractor.py
View File

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