Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-10-15 20:07:23 +02:00
Code Issues Projects Releases Wiki Activity

debugging

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-10 20:54:11 -06:00
parent ee8091ba91
commit 1ef55eb965
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -59,6 +59,7 @@ class GenericONNXEmbedding:
requires_fp16=model_size == "large" or self.model_type == "text",
openvino_device=device,
)
print(f"Detected providers {self.providers} and options {self.provider_options}")
self.download_path = os.path.join(MODEL_CACHE_DIR, self.model_name)
self.tokenizer = None
@ -131,6 +132,7 @@ class GenericONNXEmbedding:
self.tokenizer = self._load_tokenizer()
else:
self.feature_extractor = self._load_feature_extractor()
print(f"loading model {os.path.join(self.download_path, self.model_file)}")
self.session = self._load_model(
os.path.join(self.download_path, self.model_file)
)