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" )