From 09e0ef1dac30c63e6c02d1d98f67cf5386ede70b Mon Sep 17 00:00:00 2001
From: Francesco Mattioli <Francesco.mttl@gmail.com>
Date: Thu, 10 Oct 2024 18:04:04 +0200
Subject: [PATCH] fixed benchmark

---
 ultralytics/utils/benchmarks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ultralytics/utils/benchmarks.py b/ultralytics/utils/benchmarks.py
index eb60214d2e..0a6061f6f0 100644
--- 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"