From 473a061d7c777641d005f41c3c6fca4ba687dc77 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 18 Sep 2025 22:24:27 -0600 Subject: [PATCH] Formatting --- frigate/detectors/detection_runners.py | 25 +++++++++++++++---------- 1 file changed, 15 insertions(+), 10 deletions(-) diff --git a/frigate/detectors/detection_runners.py b/frigate/detectors/detection_runners.py index 20227b11d..9babbdef3 100644 --- a/frigate/detectors/detection_runners.py +++ b/frigate/detectors/detection_runners.py @@ -428,14 +428,19 @@ def get_optimized_runner( **options[0], "enable_cuda_graph": True, } - return CudaGraphRunner(ort.InferenceSession( - model_path, - providers=providers, - provider_options=options, - ), options[0]["device_id"]) + return CudaGraphRunner( + ort.InferenceSession( + model_path, + providers=providers, + provider_options=options, + ), + options[0]["device_id"], + ) - return ONNXModelRunner(ort.InferenceSession( - model_path, - providers=providers, - provider_options=options, - )) + return ONNXModelRunner( + ort.InferenceSession( + model_path, + providers=providers, + provider_options=options, + ) + )