From 629e2c59b132265dca7550d6198c9edfd8921283 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Thu, 10 Oct 2024 12:27:10 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- ultralytics/engine/exporter.py | 2 +- ultralytics/nn/autobackend.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index d26feb109d..5ef86492a8 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1041,7 +1041,7 @@ class Exporter: @try_export def export_mct(self, prefix=colorstr("Sony MCT:")): - check_requirements(["mct-nightly","sony-custom-layers[torch]"]) + check_requirements(["mct-nightly", "sony-custom-layers[torch]"]) import model_compression_toolkit as mct import onnx from model_compression_toolkit.core import BitWidthConfig diff --git a/ultralytics/nn/autobackend.py b/ultralytics/nn/autobackend.py index 57467626d4..161dd8670d 100644 --- a/ultralytics/nn/autobackend.py +++ b/ultralytics/nn/autobackend.py @@ -178,7 +178,7 @@ class AutoBackend(nn.Module): providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"] if mct: - check_requirements(["mct-nightly","sony-custom-layers[torch]"]) + check_requirements(["mct-nightly", "sony-custom-layers[torch]"]) LOGGER.info(f"Loading {w} for ONNX MCT quantization inference...") import mct_quantizers as mctq from sony_custom_layers.pytorch.object_detection import nms_ort # noqa