From 178117183ed0b43e9d7e6705ce0fbb1d5619d7c4 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Mon, 7 Jul 2025 17:57:37 -0600 Subject: [PATCH] Move trt error (#19031) * Move trt error * Formatting --- frigate/detectors/plugins/tensorrt.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index b46c24f1f..bf0eb6fa8 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -43,13 +43,6 @@ if TRT_SUPPORT: return logging.CRITICAL else: return logging.DEBUG -elif platform.machine() == "x86_64": - logger.error( - "TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information." - ) - raise ImportError( - "TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information." - ) class TensorRTDetectorConfig(BaseDetectorConfig): @@ -227,6 +220,14 @@ class TensorRtDetector(DetectionApi): ] def __init__(self, detector_config: TensorRTDetectorConfig): + if platform.machine() == "x86_64": + logger.error( + "TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information." + ) + raise ImportError( + "TensorRT detector is no longer supported on amd64 system. Please use ONNX detector instead, see https://docs.frigate.video/configuration/object_detectors#onnx for more information." + ) + assert TRT_SUPPORT, ( f"TensorRT libraries not found, {DETECTOR_KEY} detector not present" )