Merge pull request #1790 from roboflow/fix/yolo-nas-in-trt

Get rid of verification of named outputs in case of YOLO-NAS TRT models - as the artefacts are clearly inconsistent

Author: PawelPeczek-Roboflow

inference_experimental/inference_exp/models/yolonas/yolonas_object_detection_trt.py

@@ -126,15 +126,11 @@ def from_pretrained(
                 message=f"Implementation assume 2 model outputs, found: {len(outputs)}.",
                 help_url="https://todo",
             )
-        if "output0" not in outputs or "output1" not in outputs:
-            raise CorruptedModelPackageError(
-                message=f"Expected model outputs to be named `output0` and `output1`, but found: {outputs}.",
-                help_url="https://todo",
-            )
+        # git rid of outputs order and names verification, as YOLO-NAS clearly produces different outputs
         return cls(
             engine=engine,
             input_name=inputs[0],
-            output_names=["output0", "output1"],
+            output_names=outputs,
             class_names=class_names,
             inference_config=inference_config,
             trt_config=trt_config,

inference_experimental/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "inference-exp"
-version = "0.17.0rc2"
+version = "0.17.0"
 description = "Experimental vresion of inference package which is supposed to evolve into inference 1.0"
 readme = "README.md"
 requires-python = ">=3.9,<3.13"