Add yolov5 alias for yolo (#1111)

neuralmagic · Jul 19, 2023 · 550aeaa · 550aeaa
1 parent 23178bc
commit 550aeaa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -271,6 +271,7 @@ def _setup_extras() -> Dict:
         "onnxruntime": _onnxruntime_deps,
         "image_classification": _image_classification_deps,
         "yolo": _yolo_integration_deps,
+        "yolov5": _yolo_integration_deps,
         "haystack": _haystack_integration_deps,
         "openpifpaf": _openpifpaf_integration_deps,
         "yolov8": _yolov8_integration_deps,