mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-02-13 13:35:21 +03:00
bugfix: onnx inferencer nono
This commit is contained in:
parent
5a72d0a8cd
commit
75b575d81d
@ -15,12 +15,10 @@ class OnnxRVCInferencerNono(OnnxRVCInferencer):
|
|||||||
# ort_options = onnxruntime.SessionOptions()
|
# ort_options = onnxruntime.SessionOptions()
|
||||||
# ort_options.intra_op_num_threads = 8
|
# ort_options.intra_op_num_threads = 8
|
||||||
|
|
||||||
onnx_session = onnxruntime.InferenceSession(
|
onnx_session = onnxruntime.InferenceSession(file, providers=providers)
|
||||||
self.onnx_model, providers=providers
|
|
||||||
)
|
|
||||||
|
|
||||||
# check half-precision
|
# 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)":
|
if first_input_type == "tensor(float)":
|
||||||
self.isHalf = False
|
self.isHalf = False
|
||||||
else:
|
else:
|
||||||
|
Loading…
Reference in New Issue
Block a user