Skip to content

Commit fe4c4ff

Browse files
chore: Updated warning message
1 parent 2e6c8ab commit fe4c4ff

File tree

4 files changed

+14
-15
lines changed

4 files changed

+14
-15
lines changed

fastembed/text/pooled_embedding.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,17 @@
2929
},
3030
"model_file": "onnx/model.onnx",
3131
},
32+
{
33+
"model": "nomic-ai/nomic-embed-text-v1.5-Q",
34+
"dim": 768,
35+
"description": "Text embeddings, Multimodal (text, image), English, 8192 input tokens truncation, Prefixes for queries/documents: necessary, 2024 year.",
36+
"license": "apache-2.0",
37+
"size_in_GB": 0.13,
38+
"sources": {
39+
"hf": "nomic-ai/nomic-embed-text-v1.5",
40+
},
41+
"model_file": "onnx/model_quantized.onnx",
42+
},
3243
]
3344

3445

fastembed/text/pooled_normalized_embedding.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -9,17 +9,6 @@
99
from fastembed.text.pooled_embedding import PooledEmbedding
1010

1111
supported_pooled_normalized_models = [
12-
{
13-
"model": "nomic-ai/nomic-embed-text-v1.5-Q",
14-
"dim": 768,
15-
"description": "Text embeddings, Multimodal (text, image), English, 8192 input tokens truncation, Prefixes for queries/documents: necessary, 2024 year.",
16-
"license": "apache-2.0",
17-
"size_in_GB": 0.13,
18-
"sources": {
19-
"hf": "nomic-ai/nomic-embed-text-v1.5",
20-
},
21-
"model_file": "onnx/model_quantized.onnx",
22-
},
2312
{
2413
"model": "sentence-transformers/all-MiniLM-L6-v2",
2514
"dim": 384,

fastembed/text/text_embedding.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,9 +65,8 @@ def __init__(
6565
super().__init__(model_name, cache_dir, threads, **kwargs)
6666
if model_name == "nomic-ai/nomic-embed-text-v1.5-Q":
6767
warnings.warn(
68-
"The model 'nomic-ai/nomic-embed-text-v1.5-Q' has been updated to include a mean pooling layer. "
69-
"Please ensure your usage aligns with the new functionality. "
70-
"Support for the previous version without mean pooling will be removed as of version 0.5.2.",
68+
"The model 'nomic-ai/nomic-embed-text-v1.5-Q' has been updated on HuggingFace. "
69+
"Please review the latest documentation and release notes to ensure compatibility with your workflow. ",
7170
UserWarning,
7271
stacklevel=2,
7372
)

tests/test_text_onnx_embeddings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545
"jinaai/jina-embeddings-v2-base-es": np.array([-0.0108, -0.0092, -0.0373, 0.0171, -0.0301]),
4646
"nomic-ai/nomic-embed-text-v1": np.array([0.3708, 0.2031, -0.3406, -0.2114, -0.3230]),
4747
"nomic-ai/nomic-embed-text-v1.5": np.array(
48-
[-0.15407836, -0.03053198, -3.9138033, 0.1910364, 0.13224715]
48+
[0.0802303, 0.3700881, -4.3053818, 0.4431803, -0.271572]
4949
),
5050
"nomic-ai/nomic-embed-text-v1.5-Q": np.array(
5151
[0.0035279, 0.0162738, -0.1893193, 0.0194878, -0.0119418]

0 commit comments

Comments
 (0)