Update config.pbtxt

2025-09-22 08:08:23 +00:00 · 2025-09-22 08:08:23 +00:00 · c33bac8317
commit c33bac8317
parent c4da556a24
1 changed files with 12 additions and 0 deletions
--- a/config.pbtxt
+++ b/config.pbtxt
@ -7,9 +7,21 @@ max_batch_size: 0
 default_model_filename: "yolo11n.engine"

 input [
+  {
+    name: "images"
+    data_type: TYPE_FP32
+    dims: [1, 3, 640, 640]
+  }
 ]

+# Triton should expect to respond with a single string
+# output of variable length named 'text_output'
 output [
+  {
+    name: "output0"
+    data_type: TYPE_FP32
+    dims: [1, 84, 8400]
+  }
 ]