Fix yolov8 export alias (#1579)

neuralmagic · May 26, 2023 · 53e2f8d · 53e2f8d
1 parent 4f7ba12
commit 53e2f8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -249,7 +249,7 @@ def _setup_entry_points() -> Dict:
         [
             "sparseml.ultralytics.train=sparseml.yolov8.train:main",
             "sparseml.ultralytics.val=sparseml.yolov8.val:main",
-            "sparseml.ultralytics.export=sparseml.yolov8.export:main",
+            "sparseml.ultralytics.export_onnx=sparseml.yolov8.export:main",
         ]
     )