This commit is contained in:
Nicolas Mowen 2025-09-17 14:45:28 -06:00
parent d1f209e7e9
commit d1f4e73fe4

View File

@ -391,7 +391,7 @@ def get_optimized_runner(
return RKNNModelRunner(rknn_path)
if device != "CPU" and is_openvino_gpu_npu_available():
return OpenVINOModelRunner(model_path, device or "AUTO", **kwargs)
return OpenVINOModelRunner(model_path, device or "AUTO", complex_model, **kwargs)
providers, options = get_ort_providers(device == "CPU", device, **kwargs)
ortSession = ort.InferenceSession(