This commit is contained in:
w-okada 2023-09-21 09:28:41 +09:00
parent 6f8710fd50
commit 654de69533

View File

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