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, + ) + )