bugfix: onnx nof0 no work

This commit is contained in:
wataru 2023-06-05 04:08:03 +09:00
parent d6731f4adf
commit 473f02951d
3 changed files with 4 additions and 3 deletions

View File

@ -42,7 +42,6 @@ class OnnxRVCInferencer(Inferencer):
# print("INFER1", self.model.get_providers())
# print("INFER2", self.model.get_provider_options())
# print("INFER3", self.model.get_session_options())
if self.isHalf:
audio1 = self.model.run(
["audio"],

View File

@ -8,7 +8,7 @@ from voice_changer.RVC.inferencer.OnnxRVCInferencer import OnnxRVCInferencer
class OnnxRVCInferencerNono(OnnxRVCInferencer):
def loadModel(self, file: str, gpu: int):
super().loadModel(file, gpu)
self.setProps(EnumInferenceTypes.onnxRVCNono, file, True, gpu)
self.setProps(EnumInferenceTypes.onnxRVCNono, file, self.isHalf, gpu)
return self
def infer(

View File

@ -42,7 +42,9 @@ class Pipeline(object):
self.embedder = embedder
self.inferencer = inferencer
self.pitchExtractor = pitchExtractor
print("GENERATE INFERENCER:::", self.inferencer)
print("GENERATE INFERENCER", self.inferencer)
print("GENERATE EMBEDDER", self.embedder)
print("GENERATE PITCH EXTRACTOR", self.pitchExtractor)
self.index = index
self.big_npy = (