diff --git a/frigate/detectors/detection_runners.py b/frigate/detectors/detection_runners.py index f5b11cc56..3e304d63c 100644 --- a/frigate/detectors/detection_runners.py +++ b/frigate/detectors/detection_runners.py @@ -203,7 +203,7 @@ class OpenVINOModelRunner(BaseModelRunner): self.infer_request = self.compiled_model.create_infer_request() self.input_tensor: ov.Tensor | None = None - if not complex_model: + if not self.complex_model: try: input_shape = self.compiled_model.inputs[0].get_shape() input_element_type = self.compiled_model.inputs[0].get_element_type() diff --git a/frigate/embeddings/onnx/base_embedding.py b/frigate/embeddings/onnx/base_embedding.py index bd15c77a8..c0bd58475 100644 --- a/frigate/embeddings/onnx/base_embedding.py +++ b/frigate/embeddings/onnx/base_embedding.py @@ -3,7 +3,6 @@ import logging import os from abc import ABC, abstractmethod -from enum import Enum from io import BytesIO from typing import Any diff --git a/frigate/embeddings/onnx/jina_v1_embedding.py b/frigate/embeddings/onnx/jina_v1_embedding.py index 80466511d..e64d8da39 100644 --- a/frigate/embeddings/onnx/jina_v1_embedding.py +++ b/frigate/embeddings/onnx/jina_v1_embedding.py @@ -4,16 +4,17 @@ import logging import os import warnings -# importing this without pytorch or others causes a warning -# https://github.com/huggingface/transformers/issues/27214 -# suppressed by setting env TRANSFORMERS_NO_ADVISORY_WARNINGS=1 -from frigate.embeddings.types import EnrichmentModelTypeEnum from transformers import AutoFeatureExtractor, AutoTokenizer from transformers.utils.logging import disable_progress_bar from frigate.comms.inter_process import InterProcessRequestor from frigate.const import MODEL_CACHE_DIR, UPDATE_MODEL_STATE from frigate.detectors.detection_runners import BaseModelRunner, get_optimized_runner + +# importing this without pytorch or others causes a warning +# https://github.com/huggingface/transformers/issues/27214 +# suppressed by setting env TRANSFORMERS_NO_ADVISORY_WARNINGS=1 +from frigate.embeddings.types import EnrichmentModelTypeEnum from frigate.types import ModelStatusTypesEnum from frigate.util.downloader import ModelDownloader diff --git a/frigate/embeddings/onnx/jina_v2_embedding.py b/frigate/embeddings/onnx/jina_v2_embedding.py index d8fd96c50..44cc6c12b 100644 --- a/frigate/embeddings/onnx/jina_v2_embedding.py +++ b/frigate/embeddings/onnx/jina_v2_embedding.py @@ -6,13 +6,13 @@ import os import numpy as np from PIL import Image -from frigate.embeddings.types import EnrichmentModelTypeEnum from transformers import AutoTokenizer from transformers.utils.logging import disable_progress_bar, set_verbosity_error from frigate.comms.inter_process import InterProcessRequestor from frigate.const import MODEL_CACHE_DIR, UPDATE_MODEL_STATE from frigate.detectors.detection_runners import get_optimized_runner +from frigate.embeddings.types import EnrichmentModelTypeEnum from frigate.types import ModelStatusTypesEnum from frigate.util.downloader import ModelDownloader