From e8daaaea147fa51fd8c8a75afc5979c3e6417c99 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Mon, 21 Oct 2024 16:17:12 -0600 Subject: [PATCH] Formatting --- frigate/embeddings/embeddings.py | 2 +- frigate/embeddings/functions/onnx.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) 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(