Skip to content

Commit 0982dbf

Browse files
committed
fix: use original jina de model instead of fp16 due to onnxruntime updates
1 parent 21df54e commit 0982dbf

2 files changed

Lines changed: 3 additions & 17 deletions

File tree

fastembed/text/pooled_normalized_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@
5959
license="apache-2.0",
6060
size_in_GB=0.32,
6161
sources=ModelSource(hf="jinaai/jina-embeddings-v2-base-de"),
62-
model_file="onnx/model_fp16.onnx",
62+
model_file="onnx/model.onnx",
6363
),
6464
DenseModelDescription(
6565
model="jinaai/jina-embeddings-v2-base-code",

fastembed/text/text_embedding.py

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -90,23 +90,9 @@ def __init__(
9090
**kwargs: Any,
9191
):
9292
super().__init__(model_name, cache_dir, threads, **kwargs)
93-
if model_name.lower() == "nomic-ai/nomic-embed-text-v1.5-Q".lower():
93+
if model_name.lower() == "jinaai/jina-embeddings-v2-base-de":
9494
warnings.warn(
95-
"The model 'nomic-ai/nomic-embed-text-v1.5-Q' has been updated on HuggingFace. Please review "
96-
"the latest documentation on HF and release notes to ensure compatibility with your workflow. ",
97-
UserWarning,
98-
stacklevel=2,
99-
)
100-
if model_name.lower() in {
101-
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2".lower(),
102-
"thenlper/gte-large".lower(),
103-
"intfloat/multilingual-e5-large".lower(),
104-
"sentence-transformers/paraphrase-multilingual-mpnet-base-v2".lower(),
105-
}:
106-
warnings.warn(
107-
f"The model {model_name} now uses mean pooling instead of CLS embedding. "
108-
f"In order to preserve the previous behaviour, consider either pinning fastembed version to 0.5.1 or "
109-
"using `add_custom_model` functionality.",
95+
"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.",
11096
UserWarning,
11197
stacklevel=2,
11298
)

0 commit comments

Comments
 (0)