updated mct version

mct-2.1.1
Francesco Mattioli 4 months ago
parent 0a0129cd7d
commit 83e0cb5bc1
  1. 2
      ultralytics/engine/exporter.py
  2. 2
      ultralytics/nn/autobackend.py

@ -1075,7 +1075,7 @@ class Exporter:
@try_export
def export_mct(self, prefix=colorstr("Sony MCT:")):
check_requirements(["mct-nightly", "sony-custom-layers[torch]"])
check_requirements(["model-compression-toolkit==2.2.1", "sony-custom-layers[torch]"])
import subprocess
import model_compression_toolkit as mct

@ -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]", "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

Loading…
Cancel
Save