fixed benchmark

5 months ago · 09e0ef1dac
parent 3d533587a7
commit 09e0ef1dac
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/utils/benchmarks.py
+++ b/ultralytics/utils/benchmarks.py
@ -114,7 +114,7 @@ def benchmark(
            if i in {12}:  # NCNN
                assert not isinstance(model, YOLOWorld), "YOLOWorldv2 NCNN exports not supported yet"
            if i in {13}:  # MCT
-                assert not model.model.end2end
+                assert not is_end2end
                assert not isinstance(model, YOLOWorld), "YOLOWorldv2 MCT exports not supported"
                assert model.task == "detect", "MCT only supported for detection task"
                assert "C2f" in model.__str__(), "MCT only supported for YOLOv8"