From 157132973ad8dcc3c3717c9d1eb6095ed7afe44e Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 18 Sep 2025 14:13:10 -0600 Subject: [PATCH] Clenanup --- frigate/detectors/detection_runners.py | 4 ++-- frigate/embeddings/types.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/frigate/detectors/detection_runners.py b/frigate/detectors/detection_runners.py index 332cf9488..f5b11cc56 100644 --- a/frigate/detectors/detection_runners.py +++ b/frigate/detectors/detection_runners.py @@ -106,7 +106,7 @@ class CudaGraphRunner(BaseModelRunner): # Import here to avoid circular imports from frigate.detectors.detector_config import ModelTypeEnum from frigate.embeddings.types import EnrichmentModelTypeEnum - + return model_type in [ ModelTypeEnum.yolonas.value, EnrichmentModelTypeEnum.paddleocr.value, @@ -173,7 +173,7 @@ class OpenVINOModelRunner(BaseModelRunner): def is_complex_model(model_type: str) -> bool: # Import here to avoid circular imports from frigate.embeddings.types import EnrichmentModelTypeEnum - + return model_type in [EnrichmentModelTypeEnum.paddleocr.value] def __init__(self, model_path: str, device: str, model_type: str, **kwargs): diff --git a/frigate/embeddings/types.py b/frigate/embeddings/types.py index 7fd7e43fa..2ed404437 100644 --- a/frigate/embeddings/types.py +++ b/frigate/embeddings/types.py @@ -5,9 +5,10 @@ class EmbeddingTypeEnum(str, Enum): thumbnail = "thumbnail" description = "description" + class EnrichmentModelTypeEnum(str, Enum): arcface = "arcface" facenet = "facenet" jina_v1 = "jina_v1" jina_v2 = "jina_v2" - paddleocr = "paddleocr" \ No newline at end of file + paddleocr = "paddleocr"