diff --git a/frigate/detectors/plugins/onnx.py b/frigate/detectors/plugins/onnx.py index 23c3aeb20..7c7670733 100644 --- a/frigate/detectors/plugins/onnx.py +++ b/frigate/detectors/plugins/onnx.py @@ -56,7 +56,9 @@ class ONNXDetector(DetectionApi): if providers[0] == "ROCMExecutionProvider": # avoid AMD GPU kernel crashes sess_options = ort.SessionOptions() - sess_options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_DISABLE_ALL + sess_options.graph_optimization_level = ( + ort.GraphOptimizationLevel.ORT_DISABLE_ALL + ) self.model = ort.InferenceSession( path, providers=providers, provider_options=options