diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py index 388624fdb..d77a9eecf 100644 --- a/frigate/embeddings/embeddings.py +++ b/frigate/embeddings/embeddings.py @@ -20,7 +20,7 @@ from frigate.models import Event from frigate.types import ModelStatusTypesEnum from frigate.util.builtin import serialize -from .functions.onnx import GenericONNXEmbedding +from .functions.onnx import GenericONNXEmbedding, ModelTypeEnum logger = logging.getLogger(__name__) diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py index 66bdfe5e7..6ea495a30 100644 --- a/frigate/embeddings/functions/onnx.py +++ b/frigate/embeddings/functions/onnx.py @@ -95,7 +95,10 @@ class GenericONNXEmbedding: file_name = os.path.basename(path) if file_name in self.download_urls: ModelDownloader.download_from_url(self.download_urls[file_name], path) - elif file_name == self.tokenizer_file and self.model_type == ModelTypeEnum.text: + elif ( + file_name == self.tokenizer_file + and self.model_type == ModelTypeEnum.text + ): if not os.path.exists(path + "/" + self.model_name): logger.info(f"Downloading {self.model_name} tokenizer") tokenizer = AutoTokenizer.from_pretrained(