From 6eb21d7c7b1057f474caed752b0c98f0bf7ffeb5 Mon Sep 17 00:00:00 2001
From: Konstantin <46416203+TheConstant3@users.noreply.github.com>
Date: Mon, 28 Oct 2024 09:08:48 +0300
Subject: [PATCH] add TensorrtExecutionProvider to yolov8onnx

---
 sahi/models/yolov8onnx.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()