From 4bca405e29ebaf69a99ce58f32e9606bb5b9a5c2 Mon Sep 17 00:00:00 2001 From: Marc Altmann <40744649+MarcA711@users.noreply.github.com> Date: Tue, 18 Jun 2024 15:40:54 +0200 Subject: [PATCH] fix key error for custom models (#12042) --- frigate/detectors/plugins/rknn.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py index 49f1f4c02..af22ca358 100644 --- a/frigate/detectors/plugins/rknn.py +++ b/frigate/detectors/plugins/rknn.py @@ -41,12 +41,12 @@ class Rknn(DetectionApi): 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