From a6a4e375cabf352f540f7c7c3a7559aa03dba050 Mon Sep 17 00:00:00 2001 From: George Date: Wed, 15 Apr 2026 12:53:10 +0700 Subject: [PATCH] =?UTF-8?q?fix:=20use=20original=20jina=20de=20model=20ins?= =?UTF-8?q?tead=20of=20fp16=20due=20to=20onnxruntime=20up=E2=80=A6=20(#623?= =?UTF-8?q?)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * fix: use original jina de model instead of fp16 due to onnxruntime updates * fix: update size in description --- fastembed/text/pooled_normalized_embedding.py | 4 ++-- fastembed/text/text_embedding.py | 18 ++---------------- 2 files changed, 4 insertions(+), 18 deletions(-) diff --git a/fastembed/text/pooled_normalized_embedding.py b/fastembed/text/pooled_normalized_embedding.py index 5711da1..44511d2 100644 --- a/fastembed/text/pooled_normalized_embedding.py +++ b/fastembed/text/pooled_normalized_embedding.py @@ -57,9 +57,9 @@ supported_pooled_normalized_models: list[DenseModelDescription] = [ "Prefixes for queries/documents: not necessary, 2024 year." ), license="apache-2.0", - size_in_GB=0.32, + size_in_GB=0.64, sources=ModelSource(hf="jinaai/jina-embeddings-v2-base-de"), - model_file="onnx/model_fp16.onnx", + model_file="onnx/model.onnx", ), DenseModelDescription( model="jinaai/jina-embeddings-v2-base-code", diff --git a/fastembed/text/text_embedding.py b/fastembed/text/text_embedding.py index 5a37e37..51f1a2f 100644 --- a/fastembed/text/text_embedding.py +++ b/fastembed/text/text_embedding.py @@ -90,23 +90,9 @@ class TextEmbedding(TextEmbeddingBase): **kwargs: Any, ): super().__init__(model_name, cache_dir, threads, **kwargs) - if model_name.lower() == "nomic-ai/nomic-embed-text-v1.5-Q".lower(): + if model_name.lower() == "jinaai/jina-embeddings-v2-base-de": warnings.warn( - "The model 'nomic-ai/nomic-embed-text-v1.5-Q' has been updated on HuggingFace. Please review " - "the latest documentation on HF and release notes to ensure compatibility with your workflow. ", - UserWarning, - stacklevel=2, - ) - if model_name.lower() in { - "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2".lower(), - "thenlper/gte-large".lower(), - "intfloat/multilingual-e5-large".lower(), - "sentence-transformers/paraphrase-multilingual-mpnet-base-v2".lower(), - }: - warnings.warn( - f"The model {model_name} now uses mean pooling instead of CLS embedding. " - f"In order to preserve the previous behaviour, consider either pinning fastembed version to 0.5.1 or " - "using `add_custom_model` functionality.", + "The model 'jinaai/jina-embeddings-v2-base-de' used to run with fp16 model, but due to onnxruntime updates, now it runs with the original fp32 model.", UserWarning, stacklevel=2, )