diff --git a/val_onnx.py b/val_onnx.py
index eea06f55c873..9bcbf46f3f71 100644
--- a/val_onnx.py
+++ b/val_onnx.py
@@ -310,7 +310,10 @@ def run(
 
         # Inference
         out = yolo_pipeline(
-            images=[im.numpy()], iou_thres=iou_thres, conf_thres=conf_thres
+            images=[im.numpy()], 
+            iou_thres=iou_thres, 
+            conf_thres=conf_thres, 
+            multi_label=True
         )
 
         # inference, loss outputs