From 60c66f24db4bb28d326843ba0f568e060034c7e7 Mon Sep 17 00:00:00 2001 From: Francesco Mattioli Date: Mon, 16 Sep 2024 12:03:59 +0200 Subject: [PATCH] added mct dependency --- ultralytics/engine/exporter.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 85facfa9a9..3b6f3719d0 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1035,6 +1035,7 @@ class Exporter: @try_export def export_mct(self, prefix=colorstr("Sony MCT:")): # pip install --upgrade -force-reinstall git+https://github.com/ambitious-octopus/model_optimization.git@get-output-fix + check_requirements("mct-nightly") import model_compression_toolkit as mct import onnx from model_compression_toolkit.core.pytorch.pytorch_device_config import get_working_device, set_working_device