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

Merge pull request #2 from idxlics/patch-2

Browse Source 
Update jina_v2_embedding.py
This commit is contained in:
idxlics 2025-05-04 23:10:02 +08:00 committed by GitHub
commit ea51239f6c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -34,12 +34,13 @@ class JinaV2Embedding(BaseEmbedding):
model_file = ( model_file = (
"model_fp16.onnx" if model_size == "large" else "model_quantized.onnx" "model_fp16.onnx" if model_size == "large" else "model_quantized.onnx"
) )
HF_ENDPOINT = os.environ.get("HF_ENDPOINT", "https://huggingface.co")
super().__init__( super().__init__(
model_name="jinaai/jina-clip-v2", model_name="jinaai/jina-clip-v2",
model_file=model_file, model_file=model_file,
download_urls={ download_urls={
model_file: f"https://huggingface.co/jinaai/jina-clip-v2/resolve/main/onnx/{model_file}", model_file: f"{HF_ENDPOINT}/jinaai/jina-clip-v2/resolve/main/onnx/{model_file}",
"preprocessor_config.json": "https://huggingface.co/jinaai/jina-clip-v2/resolve/main/preprocessor_config.json", "preprocessor_config.json": f"{HF_ENDPOINT}/jinaai/jina-clip-v2/resolve/main/preprocessor_config.json",
}, },
) )
self.tokenizer_file = "tokenizer" self.tokenizer_file = "tokenizer"