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

pass model size

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-10 17:37:18 -05:00
parent cdb794ece8
commit 080598485f
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/embeddings/embeddings.py
View File

@ -97,6 +97,7 @@ class Embeddings:
"text_model_fp16.onnx": "https://huggingface.co/jinaai/jina-clip-v1/resolve/main/onnx/text_model_fp16.onnx",
},
embedding_function=jina_text_embedding_function,
model_size=config.model_size,
model_type="text",
requestor=self.requestor,
device="CPU",
@ -118,6 +119,7 @@ class Embeddings:
model_file=model_file,
download_urls=download_urls,
embedding_function=jina_vision_embedding_function,
model_size=config.model_size,
model_type="vision",
requestor=self.requestor,
device=self.config.device,

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

@ -41,6 +41,7 @@ class GenericONNXEmbedding:
model_file: str,
download_urls: Dict[str, str],
embedding_function: Callable[[List[np.ndarray]], np.ndarray],
model_size: str,
model_type: str,
requestor: InterProcessRequestor,
tokenizer_file: Optional[str] = None,
@ -55,8 +56,7 @@ class GenericONNXEmbedding:
self.model_type = model_type # 'text' or 'vision'
self.providers, self.provider_options = get_ort_providers(
force_cpu=device == "CPU",
requires_fp16=self.config.model_size == "large"
or self.model_type == "text",
requires_fp16=model_size == "large" or self.model_type == "text",
openvino_device=device,
)