From 95641d09819d990195d0070a323bc2cc62ab108f Mon Sep 17 00:00:00 2001 From: stephantul Date: Thu, 27 Aug 2026 11:27:07 +0200 Subject: [PATCH 1/7] new: add minish models --- fastembed/text/onnx_embedding.py | 31 +++++++++++++++++++++++++++++-- 1 file changed, 29 insertions(+), 2 deletions(-) diff --git a/fastembed/text/onnx_embedding.py b/fastembed/text/onnx_embedding.py index 1e9978f75..577d59498 100644 --- a/fastembed/text/onnx_embedding.py +++ b/fastembed/text/onnx_embedding.py @@ -1,11 +1,11 @@ from typing import Any, Iterable, Sequence, Type -from fastembed.common.types import NumpyArray, OnnxProvider, Device +from fastembed.common.model_description import DenseModelDescription, ModelSource from fastembed.common.onnx_model import OnnxOutputContext +from fastembed.common.types import Device, NumpyArray, OnnxProvider from fastembed.common.utils import define_cache_dir, normalize from fastembed.text.onnx_text_model import OnnxTextModel, TextEmbeddingWorker from fastembed.text.text_embedding_base import TextEmbeddingBase -from fastembed.common.model_description import DenseModelDescription, ModelSource supported_onnx_models: list[DenseModelDescription] = [ DenseModelDescription( @@ -180,6 +180,33 @@ sources=ModelSource(hf="jinaai/jina-clip-v1"), model_file="onnx/text_model.onnx", ), + DenseModelDescription( + model="minishlab/potion-base-8m", + dim=256, + description=("Text embeddings, Unimodal (text), English, 2024 year."), + license="mit", + size_in_GB=0.030, + sources=ModelSource(hf="minishlab/potion-base-8m-onnx"), + model_file="model.onnx", + ), + DenseModelDescription( + model="minishlab/potion-retrieval-32m", + dim=512, + description=("Text embeddings, Unimodal (text), English, 2025 year."), + license="mit", + size_in_GB=0.129, + sources=ModelSource(hf="minishlab/potion-retrieval-32m-onnx"), + model_file="model.onnx", + ), + DenseModelDescription( + model="minishlab/potion-multilingual-128m", + dim=256, + description=("Text embeddings, Unimodal (text), Multilingual, 2025 year."), + license="mit", + size_in_GB=0.512, + sources=ModelSource(hf="minishlab/potion-multilingual-128m-onnx"), + model_file="model.onnx", + ), ] From 89020d6081113e145a3d3ed0b191b1a8f576183e Mon Sep 17 00:00:00 2001 From: stephantul Date: Thu, 27 Aug 2026 11:41:53 +0200 Subject: [PATCH 2/7] add canonical values to test --- tests/test_text_onnx_embeddings.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_text_onnx_embeddings.py b/tests/test_text_onnx_embeddings.py index 8744b6177..84b3b68d5 100644 --- a/tests/test_text_onnx_embeddings.py +++ b/tests/test_text_onnx_embeddings.py @@ -82,6 +82,15 @@ "google/siglip2-base-patch16-224": np.array( [-0.01181389, 0.00737596, 0.01118064, 0.0103095, 0.3451049] ), + "minishlab/potion-base-8m": np.array( + [-0.03432461, -0.08020256, -0.14396408, 0.08480079, 0.01958815] + ), + "minishlab/potion-retrieval-32m": np.array( + [0.019733, -0.01530093, -0.08678473, 0.0229059, 0.04700558] + ), + "minishlab/potion-multilingual-128m": np.array( + [0.02366836, 0.02973341, 0.05140258, -0.00745248, -0.06740689] + ), } QWEN3_INSTRUCT_PREFIX = ( From d8e4b494e977db870f9a4cbf4b70ff8f1f16db24 Mon Sep 17 00:00:00 2001 From: stephantul Date: Thu, 27 Aug 2026 11:43:07 +0200 Subject: [PATCH 3/7] amend description --- fastembed/text/onnx_embedding.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fastembed/text/onnx_embedding.py b/fastembed/text/onnx_embedding.py index 577d59498..97237adce 100644 --- a/fastembed/text/onnx_embedding.py +++ b/fastembed/text/onnx_embedding.py @@ -183,7 +183,7 @@ DenseModelDescription( model="minishlab/potion-base-8m", dim=256, - description=("Text embeddings, Unimodal (text), English, 2024 year."), + description=("Text embeddings, Extremely fast, Unimodal (text), English, 2024 year."), license="mit", size_in_GB=0.030, sources=ModelSource(hf="minishlab/potion-base-8m-onnx"), @@ -192,7 +192,7 @@ DenseModelDescription( model="minishlab/potion-retrieval-32m", dim=512, - description=("Text embeddings, Unimodal (text), English, 2025 year."), + description=("Text embeddings, Extremely fast, Unimodal (text), English, 2025 year."), license="mit", size_in_GB=0.129, sources=ModelSource(hf="minishlab/potion-retrieval-32m-onnx"), @@ -201,7 +201,7 @@ DenseModelDescription( model="minishlab/potion-multilingual-128m", dim=256, - description=("Text embeddings, Unimodal (text), Multilingual, 2025 year."), + description=("Text embeddings, Extremely fast, Unimodal (text), Multilingual, 2025 year."), license="mit", size_in_GB=0.512, sources=ModelSource(hf="minishlab/potion-multilingual-128m-onnx"), From 1616a731b92ecf4ca60884e5cdee1fb6d7aa5035 Mon Sep 17 00:00:00 2001 From: Stephan Tulkens Date: Fri, 28 Aug 2026 06:06:52 +0200 Subject: [PATCH 4/7] Apply suggestions from code review Co-authored-by: Dylan Couzon --- fastembed/text/onnx_embedding.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/fastembed/text/onnx_embedding.py b/fastembed/text/onnx_embedding.py index 97237adce..e188191cf 100644 --- a/fastembed/text/onnx_embedding.py +++ b/fastembed/text/onnx_embedding.py @@ -183,25 +183,34 @@ DenseModelDescription( model="minishlab/potion-base-8m", dim=256, - description=("Text embeddings, Extremely fast, Unimodal (text), English, 2024 year."), + description=( + "Text embeddings, Unimodal (text), English, 512 input tokens truncation, " + "Prefixes for queries/documents: not necessary, 2024 year." + ), license="mit", size_in_GB=0.030, sources=ModelSource(hf="minishlab/potion-base-8m-onnx"), model_file="model.onnx", ), DenseModelDescription( - model="minishlab/potion-retrieval-32m", + model="minishlab/potion-retrieval-32M", dim=512, - description=("Text embeddings, Extremely fast, Unimodal (text), English, 2025 year."), + description=( + "Text embeddings, Unimodal (text), English, 512 input tokens truncation, " + "Prefixes for queries/documents: not necessary, 2025 year." + ), license="mit", size_in_GB=0.129, sources=ModelSource(hf="minishlab/potion-retrieval-32m-onnx"), model_file="model.onnx", ), DenseModelDescription( - model="minishlab/potion-multilingual-128m", + model="minishlab/potion-multilingual-128M", dim=256, - description=("Text embeddings, Extremely fast, Unimodal (text), Multilingual, 2025 year."), + description=( + "Text embeddings, Unimodal (text), English, 512 input tokens truncation, " + "Prefixes for queries/documents: not necessary, 2025 year." + ), license="mit", size_in_GB=0.512, sources=ModelSource(hf="minishlab/potion-multilingual-128m-onnx"), From bc13dbf86d3c04f938f7bf67ffd53e361ae26c7d Mon Sep 17 00:00:00 2001 From: Stephan Tulkens Date: Fri, 28 Aug 2026 06:07:06 +0200 Subject: [PATCH 5/7] Update fastembed/text/onnx_embedding.py Co-authored-by: Dylan Couzon --- fastembed/text/onnx_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastembed/text/onnx_embedding.py b/fastembed/text/onnx_embedding.py index e188191cf..4314ebb90 100644 --- a/fastembed/text/onnx_embedding.py +++ b/fastembed/text/onnx_embedding.py @@ -181,7 +181,7 @@ model_file="onnx/text_model.onnx", ), DenseModelDescription( - model="minishlab/potion-base-8m", + model="minishlab/potion-base-8M", dim=256, description=( "Text embeddings, Unimodal (text), English, 512 input tokens truncation, " From 874e7ee584c909d86bebf6e36efe91f1a7a2c5ed Mon Sep 17 00:00:00 2001 From: stephantul Date: Fri, 28 Aug 2026 06:27:29 +0200 Subject: [PATCH 6/7] fix typo --- fastembed/text/onnx_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastembed/text/onnx_embedding.py b/fastembed/text/onnx_embedding.py index 4314ebb90..0d36e118f 100644 --- a/fastembed/text/onnx_embedding.py +++ b/fastembed/text/onnx_embedding.py @@ -208,7 +208,7 @@ model="minishlab/potion-multilingual-128M", dim=256, description=( - "Text embeddings, Unimodal (text), English, 512 input tokens truncation, " + "Text embeddings, Unimodal (text), Multilingual, 512 input tokens truncation, " "Prefixes for queries/documents: not necessary, 2025 year." ), license="mit", From d58c9002adef56d098d2f4923b0927b79b55f577 Mon Sep 17 00:00:00 2001 From: stephantul Date: Fri, 28 Aug 2026 16:21:50 +0200 Subject: [PATCH 7/7] fix keys in tests --- tests/test_text_onnx_embeddings.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_text_onnx_embeddings.py b/tests/test_text_onnx_embeddings.py index 84b3b68d5..477655a83 100644 --- a/tests/test_text_onnx_embeddings.py +++ b/tests/test_text_onnx_embeddings.py @@ -82,13 +82,13 @@ "google/siglip2-base-patch16-224": np.array( [-0.01181389, 0.00737596, 0.01118064, 0.0103095, 0.3451049] ), - "minishlab/potion-base-8m": np.array( + "minishlab/potion-base-8M": np.array( [-0.03432461, -0.08020256, -0.14396408, 0.08480079, 0.01958815] ), - "minishlab/potion-retrieval-32m": np.array( + "minishlab/potion-retrieval-32M": np.array( [0.019733, -0.01530093, -0.08678473, 0.0229059, 0.04700558] ), - "minishlab/potion-multilingual-128m": np.array( + "minishlab/potion-multilingual-128M": np.array( [0.02366836, 0.02973341, 0.05140258, -0.00745248, -0.06740689] ), }