From eca75be2c56a5a1d159fbdbd95733022eef5582e Mon Sep 17 00:00:00 2001 From: Aryan-SINGH-GIT Date: Fri, 4 Sep 2026 02:51:50 +0530 Subject: [PATCH 1/2] feat: Add ibm-granite/granite-embedding-small-english-r2 model --- fastembed/text/builtin_sentence_embedding.py | 15 +++++++++++++++ tests/test_text_onnx_embeddings.py | 3 +++ 2 files changed, 18 insertions(+) diff --git a/fastembed/text/builtin_sentence_embedding.py b/fastembed/text/builtin_sentence_embedding.py index 430ea7ca..6f26a7a9 100644 --- a/fastembed/text/builtin_sentence_embedding.py +++ b/fastembed/text/builtin_sentence_embedding.py @@ -8,6 +8,21 @@ supported_builtin_sentence_embedding_models: list[DenseModelDescription] = [ + DenseModelDescription( + model="ibm-granite/granite-embedding-small-english-r2", + dim=384, + description=( + "Text embeddings, Unimodal (text), English, 512 input tokens truncation, " + "Granite small english r2 model. 2024 year." + ), + license="apache-2.0", + size_in_GB=0.18, + sources=ModelSource( + hf="onnx-community/granite-embedding-small-english-r2-ONNX", + ), + model_file="onnx/model.onnx", + additional_files=["onnx/model.onnx_data"], + ), DenseModelDescription( model="google/embeddinggemma-300m", dim=768, diff --git a/tests/test_text_onnx_embeddings.py b/tests/test_text_onnx_embeddings.py index 8744b617..e0dd2317 100644 --- a/tests/test_text_onnx_embeddings.py +++ b/tests/test_text_onnx_embeddings.py @@ -31,6 +31,9 @@ "BAAI/bge-large-en-v1.5-quantized": np.array( [0.03434538, 0.03316108, 0.02191251, -0.03713358, -0.01577825] ), + "ibm-granite/granite-embedding-small-english-r2": np.array( + [0.47021756, -0.08181943, -0.97021246, 0.10116885, -0.16487208] + ), "sentence-transformers/all-MiniLM-L6-v2": np.array( [-0.034478, 0.03102, 0.00673, 0.02611, -0.039362] ), From 7e47ccbeea6eb2b777371a16b0fce1acbf0dc78c Mon Sep 17 00:00:00 2001 From: Aryan-SINGH-GIT Date: Fri, 4 Sep 2026 03:35:42 +0530 Subject: [PATCH 2/2] fix: update granite metadata to 8192 tokens and 2025 release year --- fastembed/text/builtin_sentence_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastembed/text/builtin_sentence_embedding.py b/fastembed/text/builtin_sentence_embedding.py index 6f26a7a9..c553c63e 100644 --- a/fastembed/text/builtin_sentence_embedding.py +++ b/fastembed/text/builtin_sentence_embedding.py @@ -12,8 +12,8 @@ model="ibm-granite/granite-embedding-small-english-r2", dim=384, description=( - "Text embeddings, Unimodal (text), English, 512 input tokens truncation, " - "Granite small english r2 model. 2024 year." + "Text embeddings, Unimodal (text), English, 8192 input tokens truncation, " + "Granite small english r2 model. 2025 year." ), license="apache-2.0", size_in_GB=0.18,