From f6005a916694dbe45657abbb84b13162a6a2b5cf Mon Sep 17 00:00:00 2001
From: Laughing <61612323+Laughing-q@users.noreply.github.com>
Date: Fri, 6 Dec 2024 18:46:48 +0800
Subject: [PATCH] Set `task` for Triton inferenceTriton (#18064)

---
 ultralytics/engine/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ultralytics/engine/model.py b/ultralytics/engine/model.py
index 874613d2..db8d87eb 100644
--- a/ultralytics/engine/model.py
+++ b/ultralytics/engine/model.py
@@ -136,6 +136,7 @@ class Model(nn.Module):
         # Check if Triton Server model
         elif self.is_triton_model(model):
             self.model_name = self.model = model
+            self.overrides["task"] = task or "detect"  # set `task=detect` if not explicitly set
             return
 
         # Load or create new YOLO model