From e0ba4c86a5e3e9f8f16e00fefdd1f63ae62b5927 Mon Sep 17 00:00:00 2001 From: Satyam Yadav <114805086+Coding-Professional@users.noreply.github.com> Date: Sat, 3 Oct 2026 02:32:59 +0530 Subject: [PATCH] feat(text): support Granite 107M multilingual embeddings --- fastembed/text/onnx_embedding.py | 12 ++++++++++++ tests/test_text_onnx_embeddings.py | 3 +++ 2 files changed, 15 insertions(+) diff --git a/fastembed/text/onnx_embedding.py b/fastembed/text/onnx_embedding.py index c112218a..1f2fd774 100644 --- a/fastembed/text/onnx_embedding.py +++ b/fastembed/text/onnx_embedding.py @@ -104,6 +104,18 @@ sources=ModelSource(hf="mixedbread-ai/mxbai-embed-large-v1"), model_file="onnx/model.onnx", ), + DenseModelDescription( + model="ibm-granite/granite-embedding-107m-multilingual", + dim=384, + description=( + "Text embeddings, Unimodal (text), Multilingual (12 languages), 512 input tokens truncation, " + "Prefixes for queries/documents: not necessary, 2024 year." + ), + license="apache-2.0", + size_in_GB=0.43, + sources=ModelSource(hf="ibm-granite/granite-embedding-107m-multilingual"), + model_file="model.onnx", + ), DenseModelDescription( model="snowflake/snowflake-arctic-embed-xs", dim=384, diff --git a/tests/test_text_onnx_embeddings.py b/tests/test_text_onnx_embeddings.py index 5110a300..3730f002 100644 --- a/tests/test_text_onnx_embeddings.py +++ b/tests/test_text_onnx_embeddings.py @@ -36,6 +36,9 @@ "ibm-granite/granite-embedding-small-english-r2": np.array( [0.47021756, -0.08181943, -0.97021246, 0.10116885, -0.16487208] ), + "ibm-granite/granite-embedding-107m-multilingual": np.array( + [0.03108855, 0.02688832, 0.08992623, 0.05844157, 0.03039573] + ), "sentence-transformers/all-MiniLM-L6-v2": np.array( [-0.034478, 0.03102, 0.00673, 0.02611, -0.039362] ),