diff --git a/server/voice_changer/RVC/pitchExtractor/PitchExtractorManager.py b/server/voice_changer/RVC/pitchExtractor/PitchExtractorManager.py index 6d3401b7..6df0e894 100644 --- a/server/voice_changer/RVC/pitchExtractor/PitchExtractorManager.py +++ b/server/voice_changer/RVC/pitchExtractor/PitchExtractorManager.py @@ -5,8 +5,8 @@ from voice_changer.RVC.pitchExtractor.DioPitchExtractor import DioPitchExtractor from voice_changer.RVC.pitchExtractor.HarvestPitchExtractor import HarvestPitchExtractor from voice_changer.RVC.pitchExtractor.CrepePitchExtractor import CrepePitchExtractor from voice_changer.RVC.pitchExtractor.PitchExtractor import PitchExtractor +from voice_changer.RVC.pitchExtractor.RMVPEOnnxPitchExtractor import RMVPEOnnxPitchExtractor from voice_changer.RVC.pitchExtractor.RMVPEPitchExtractor import RMVPEPitchExtractor -from voice_changer.RVC.pitchExtractor.RMVPOnnxEPitchExtractor import RMVPOnnxEPitchExtractor from voice_changer.utils.VoiceChangerParams import VoiceChangerParams @@ -42,7 +42,7 @@ class PitchExtractorManager(Protocol): elif pitchExtractorType == "rmvpe": return RMVPEPitchExtractor(cls.params.rmvpe, gpu) elif pitchExtractorType == "rmvpe_onnx": - return RMVPOnnxEPitchExtractor(cls.params.rmvpe_onnx, gpu) + return RMVPEOnnxPitchExtractor(cls.params.rmvpe_onnx, gpu) else: # return hubert as default print("[Voice Changer] PitchExctractor not found", pitchExtractorType)