From c9e517e7c79904415d12aeeb6abc327080cfbe6b Mon Sep 17 00:00:00 2001 From: idxlics <9525979+idxlics@users.noreply.github.com> Date: Sun, 4 May 2025 23:01:44 +0800 Subject: [PATCH] Update jina_v2_embedding.py --- frigate/embeddings/onnx/jina_v2_embedding.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/frigate/embeddings/onnx/jina_v2_embedding.py b/frigate/embeddings/onnx/jina_v2_embedding.py index be6573e50..e9def9a07 100644 --- a/frigate/embeddings/onnx/jina_v2_embedding.py +++ b/frigate/embeddings/onnx/jina_v2_embedding.py @@ -34,12 +34,13 @@ class JinaV2Embedding(BaseEmbedding): model_file = ( "model_fp16.onnx" if model_size == "large" else "model_quantized.onnx" ) + HF_ENDPOINT = os.environ.get("HF_ENDPOINT", "https://huggingface.co") super().__init__( model_name="jinaai/jina-clip-v2", model_file=model_file, download_urls={ - model_file: f"https://huggingface.co/jinaai/jina-clip-v2/resolve/main/onnx/{model_file}", - "preprocessor_config.json": "https://huggingface.co/jinaai/jina-clip-v2/resolve/main/preprocessor_config.json", + model_file: f"{HF_ENDPOINT}/jinaai/jina-clip-v2/resolve/main/onnx/{model_file}", + "preprocessor_config.json": f"{HF_ENDPOINT}/jinaai/jina-clip-v2/resolve/main/preprocessor_config.json", }, ) self.tokenizer_file = "tokenizer"