From 51842d7b017db073cae0f06f264aececef79cb84 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Tue, 27 Aug 2024 12:55:14 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- ultralytics/nn/autobackend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/nn/autobackend.py b/ultralytics/nn/autobackend.py index 1ddc74c81..75bbbb335 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: import mct_quantizers as mctq - from sony_custom_layers.pytorch.object_detection import nms_ort + session = onnxruntime.InferenceSession( w, mctq.get_ort_session_options(), providers=["CPUExecutionProvider"] )