Back to Sentence Transformers

Pretrained Models

docs/multi_vector_encoder/pretrained_models.md

6.0.09.2 KB
Original Source

Pretrained Models

{eval-rst}
The `sentence-transformers tag <https://huggingface.co/models?library=sentence-transformers&other=multi-vector>`_
on the Hugging Face Hub is the list that stays current, and we are working to get it onto every model that works
with :class:`~sentence_transformers.multi_vector_encoder.model.MultiVectorEncoder`. The tables below are what we test against directly, so
treat them as a starting point rather than the full set. For text retrieval in particular, any PyLate or
Stanford-NLP ColBERT checkpoint loads whether or not it carries the tag yet.

Models integrate seamlessly with this simple interface:
python
from sentence_transformers import MultiVectorEncoder

# Download from the 🤗 Hub
model = MultiVectorEncoder("lightonai/LateOn")

# Run inference
queries = ["What is the capital of France?"]
documents = [
    "Paris is the capital of France.",
    "Berlin is the capital of Germany.",
]
query_embeddings = model.encode_query(queries)
document_embeddings = model.encode_document(documents)
print(query_embeddings[0].shape, document_embeddings[0].shape)
# (10, 128) (9, 128) - one 128-dimensional vector per token

# Get the late-interaction (MaxSim) similarity scores for the embeddings
similarities = model.similarity(query_embeddings, document_embeddings)
print(similarities)
# tensor([[9.1129, 8.8769]])

Text Retrieval Models

These load with their trained prefix tokens, query expansion, and punctuation skiplist recovered from the saved configuration. Where a revision is listed, pass it until the pull request on that repository is merged, after which the plain model name is enough.

The NanoBEIR column reports the mean NDCG@10 (higher is better) across the 13 NanoBEIR datasets, each a 50-query subsample of a BEIR dataset, as a fast proxy for English text retrieval quality. We used the MultiVectorNanoBEIREvaluator to compute the scores for the primarily-English models. A - means the model was not evaluated on it. Note that NanoBEIR is a small benchmark, and its scores aren't a substitute for evaluating on your own data, which is always the right way to pick a model.

ModelParametersDimensionalityNanoBEIRNotes
lightonai/LateOn-regularized149M1280.6897-
lightonai/LateOn-hpool-regularized149M1280.6876-
lightonai/LateOn149M1280.6868-
LiquidAI/LFM2.5-ColBERT-350M353M1280.6864needs trust_remote_code=True
lightonai/mLateOn307M1280.6851-
lightonai/GTE-ModernColBERT-v1149M1280.6720-
topk-io/Iso-ModernColBERT149M1280.6687-
perplexity-ai/pplx-embed-v1-late-0.6b596M1280.6662needs trust_remote_code=True
lightonai/ColBERT-Zero149M1280.6569-
answerdotai/answerai-colbert-small-v133M960.6550-
mixedbread-ai/mxbai-edge-colbert-v0-32m32M640.6524-
LiquidAI/LFM2-ColBERT-350M353M1280.6441-
mixedbread-ai/mxbai-edge-colbert-v0-17m17M480.6407-
lightonai/colbertv2.0110M1280.6201-
lightonai/LateOn-Code149M1280.6169-
lightonai/Agent-ModernColBERT149M1280.6164-
lightonai/Reason-ModernColBERT149M1280.6078-
colbert-ir/colbertv2.0110M1280.6053-
VAGOsolutions/SauerkrautLM-EuroColBERT212M1280.5982-
antoinelouis/colbert-xm853M1280.5915-
VAGOsolutions/SauerkrautLM-Multi-ModernColBERT149M1280.5886-
mixedbread-ai/mxbai-colbert-large-v1335M1280.5733revision="refs/pr/4"
lightonai/LateOn-Code-edge17M480.5274-
VAGOsolutions/SauerkrautLM-Multi-Reason-ModernColBERT149M1280.5267-
VAGOsolutions/SauerkrautLM-Reason-EuroColBERT212M1280.4479-
NeuML/biomedbert-base-colbert110M1280.4320-
yjoonjang/colbert-ko-v1149M128--
ytu-ce-cosmos/turkish-colbert111M256--
samheym/GerColBERT110M128--

Visual Document Retrieval Models

ColPali-style models embed page images as documents and text as queries.

The NanoViDoRe column reports the mean NDCG@10 (higher is better) across NanoViDoRe v3, a compact visual document retrieval benchmark spanning 8 subsets (computer science, energy, finance in English and French, HR, industrial, pharmaceuticals, and physics). Like with NanoBEIR, NanoViDoRe is a small benchmark which shouldn't replace evaluation on your own data.

ModelParametersDimensionalityNanoViDoReNotes
webAI-Official/webAI-ColVec1.1-8b8.4B6400.6580needs trust_remote_code=True
webAI-Official/webAI-ColVec1.1-4b4.5B6400.6520needs trust_remote_code=True
tencent/EVIE-Preview-4.5B4.54B1280.6405-
TomoroAI/tomoro-colqwen3-embed-8b8.8B3200.6206needs trust_remote_code=True
TomoroAI/tomoro-colqwen3-embed-4b4.4B3200.6019needs trust_remote_code=True
vidore/colqwen2.5-v0.23.8B1280.5402-
vidore/colqwen2.5-v0.13.8B1280.5395-
vidore/colqwen-omni-v0.14.4B1280.5309-
vidore/colpali-v1.32.9B1280.4802-
vidore/colpali-v1.3-hf2.9B1280.4793-
vidore/colpali-v1.22.9B1280.4691-
vidore/colqwen2-v1.02.2B1280.4685-
vidore/colqwen2-v0.12.2B1280.4526-
vidore/colpali2.9B1280.4516-
vidore/colpali-v1.12.9B1280.4314-
vidore/colsmolvlm-v0.12.1B1280.4054-
vidore/colpali-hard-v1.12.9B1280.3949-
vidore/colSmol-500M507M1280.3459-
vidore/colSmol-256M256M1280.2673-
ModernVBERT/colmodernvbert252M1280.2632-
vidore/colpali-v1.2-hf2.9B128--
vidore/colqwen2-v1.0-hf2.2B128--

Most of these are LoRA adapter repositories, with the adapter applied directly onto its base at load time. Some also have a -merged sibling on the Hub (e.g. vidore/colpali-v1.3-merged) with the adapter already folded into the weights.

The three -hf entries are the transformers-native *ForRetrieval ports. They load without any configuration, but use more modeling from transformers and less from sentence_transformers. Generally, it's preferable to use the original models instead, as the ports score approximately the same.