administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Don't run download

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-10 13:09:31 -06:00
parent 6287253563
commit 8f5e05fc71
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/embeddings/functions/onnx.py
View File

@ -59,6 +59,7 @@ class GenericONNXEmbedding:
self.feature_extractor = None self.feature_extractor = None
self.session = None self.session = None
if not all(os.path.exists(os.path.join(self.download_path, n)) for n in self.download_urls.keys()):
print("starting model download") print("starting model download")
self.downloader = ModelDownloader( self.downloader = ModelDownloader(
model_name=self.model_name, model_name=self.model_name,
@ -68,6 +69,9 @@ class GenericONNXEmbedding:
download_func=self._download_model, download_func=self._download_model,
) )
self.downloader.ensure_model_files() self.downloader.ensure_model_files()
else:
self.downloader = None
print("models are already downloaded")
def _download_model(self, path: str): def _download_model(self, path: str):
try: try:
@ -104,6 +108,7 @@ class GenericONNXEmbedding:
def _load_model_and_tokenizer(self): def _load_model_and_tokenizer(self):
if self.session is None: if self.session is None:
if self.downloader:
self.downloader.wait_for_download() self.downloader.wait_for_download()
if self.model_type == "text": if self.model_type == "text":
self.tokenizer = self._load_tokenizer() self.tokenizer = self._load_tokenizer()