Auto-format by https://ultralytics.com/actions

4 months ago · 729a5653bc
parent 83e0cb5bc1
commit 729a5653bc
1 changed files with 3 additions and 1 deletions
--- a/ultralytics/nn/autobackend.py
+++ b/ultralytics/nn/autobackend.py
@ -178,7 +178,9 @@ class AutoBackend(nn.Module):

            providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"]
            if mct:
-                check_requirements(["model-compression-toolkit==2.2.1", "sony-custom-layers[torch]", "onnxruntime-extensions"])
+                check_requirements(
+                    ["model-compression-toolkit==2.2.1", "sony-custom-layers[torch]", "onnxruntime-extensions"]
+                )
                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