bugfix: onnx inferencer nono

This commit is contained in:
wataru 2023-05-09 19:20:07 +09:00
parent 5a72d0a8cd
commit 75b575d81d

View File

@ -15,12 +15,10 @@ class OnnxRVCInferencerNono(OnnxRVCInferencer):
# ort_options = onnxruntime.SessionOptions()
# ort_options.intra_op_num_threads = 8
onnx_session = onnxruntime.InferenceSession(
self.onnx_model, providers=providers
)
onnx_session = onnxruntime.InferenceSession(file, providers=providers)
# check half-precision
first_input_type = self.onnx_session.get_inputs()[0].type
first_input_type = onnx_session.get_inputs()[0].type
if first_input_type == "tensor(float)":
self.isHalf = False
else: