fix key error for custom models (#12042)

blakeblackshear · Jun 18, 2024 · 4bca405 · 4bca405
1 parent bdda89b
commit 4bca405
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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