diff --git a/sahi/models/yolov8onnx.py b/sahi/models/yolov8onnx.py index 340eb07d..e38386f6 100644 --- a/sahi/models/yolov8onnx.py +++ b/sahi/models/yolov8onnx.py @@ -42,7 +42,7 @@ def load_model(self, ort_session_kwargs: Optional[dict] = {}) -> None: if self.device == torch.device("cpu"): EP_list = ["CPUExecutionProvider"] else: - EP_list = ["CUDAExecutionProvider"] + EP_list = ["TensorrtExecutionProvider", "CUDAExecutionProvider"] options = onnxruntime.SessionOptions()