Add YOLO-NAS aliases to inference_sdk

roboflow · Feb 7, 2024 · 83610b8 · 83610b8
1 parent da159c0
commit 83610b8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/inference_sdk/http/utils/aliases.py b/inference_sdk/http/utils/aliases.py
@@ -11,6 +11,9 @@
     "yolov8l-1280": "coco/12",
     "yolov8x-640": "coco/5",
     "yolov8x-1280": "coco/13",
+    "yolo-nas-s-640": "coco/14",
+    "yolo-nas-m-640": "coco/15",
+    "yolo-nas-l-640": "coco/16",
     "yolov8n-seg-640": "coco-dataset-vdnr1/2",
     "yolov8n-seg-1280": "coco-dataset-vdnr1/7",
     "yolov8s-seg-640": "coco-dataset-vdnr1/4",