diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py index 49f1f4c02a..af22ca3588 100644 --- a/frigate/detectors/plugins/rknn.py +++ b/frigate/detectors/plugins/rknn.py @@ -41,12 +41,12 @@ def __init__(self, config: RknnDetectorConfig): if model_props["preset"]: config.model.model_type = model_props["model_type"] - if model_props["model_type"] == ModelTypeEnum.yolonas: - logger.info( - "You are using yolo-nas with weights from DeciAI. " - "These weights are subject to their license and can't be used commercially. " - "For more information, see: https://docs.deci.ai/super-gradients/latest/LICENSE.YOLONAS.html" - ) + if model_props["model_type"] == ModelTypeEnum.yolonas: + logger.info( + "You are using yolo-nas with weights from DeciAI. " + "These weights are subject to their license and can't be used commercially. " + "For more information, see: https://docs.deci.ai/super-gradients/latest/LICENSE.YOLONAS.html" + ) from rknnlite.api import RKNNLite