add TensorrtExecutionProvider to yolov8onnx

obss · Oct 28, 2024 · 6eb21d7 · 6eb21d7
1 parent 3c8f593
commit 6eb21d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()