v1.5.2: SynapseAI v1.19.0

You can pull the Docker image associated to this release with:

docker pull ghcr.io/huggingface/tei-gaudi:1.5.2

SynapseAI v1.19.0

The codebase is validated with SynapseAI 1.19.0 and optimum-habana 1.15.0.

Validated Models

Text Embeddings

tei-gaudi currently supports Nomic, BERT, CamemBERT, XLM-RoBERTa models with absolute positions, JinaBERT model with Alibi positions and Mistral, Alibaba GTE and Qwen2 models with Rope positions.

Below are some examples of our validated models:

Architecture	Pooling	Models
BERT	Cls/Mean/Last token	BAAI/bge-large-en-v1.5 sentence-transformers/all-MiniLM-L6-v2 sentence-transformers/all-MiniLM-L12-v2 sentence-transformers/multi-qa-MiniLM-L6-cos-v1 sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 sentence-transformers/paraphrase-MiniLM-L3-v2
BERT	Splade	naver/efficient-splade-VI-BT-large-query
MPNet	Cls/Mean/Last token	sentence-transformers/all-mpnet-base-v2 sentence-transformers/paraphrase-multilingual-mpnet-base-v2 sentence-transformers/multi-qa-mpnet-base-dot-v1
ALBERT	Cls/Mean/Last token	sentence-transformers/paraphrase-albert-small-v2
Mistral	Cls/Mean/Last token	intfloat/e5-mistral-7b-instruct Salesforce/SFR-Embedding-2_R
GTE	Cls/Mean/Last token	Alibaba-NLP/gte-large-en-v1.5
JinaBERT	Cls/Mean/Last token	jinaai/jina-embeddings-v2-base-en

Sequence Classification and Re-Ranking

tei-gaudi currently supports CamemBERT, and XLM-RoBERTa Sequence Classification models with absolute positions.

Below are some examples of the currently supported models:

Task	Model Type	Model ID
Re-Ranking	XLM-RoBERTa	BAAI/bge-reranker-large
Re-Ranking	XLM-RoBERTa	BAAI/bge-reranker-base
Sentiment Analysis	RoBERTa	SamLowe/roberta-base-go_emotions

Full Changelog: huggingface/tgi-gaudi@v1.5.1...v1.5.2

v1.5.1: SynapseAI v1.18.0

You can pull the Docker image associated to this release with:

docker pull ghcr.io/huggingface/tei-gaudi:1.5.1

SynapseAI v1.18.0

The codebase is validated with SynapseAI 1.18.0 and optimum-habana 1.14.1.

Validated Models

Architecture	Model Type	Models
BERT	Embedding	BAAI/bge-large-en-v1.5 sentence-transformers/all-MiniLM-L6-v2 sentence-transformers/all-MiniLM-L12-v2 sentence-transformers/multi-qa-MiniLM-L6-cos-v1 sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 sentence-transformers/paraphrase-MiniLM-L3-v2
MPNet	Embedding	sentence-transformers/all-mpnet-base-v2 sentence-transformers/paraphrase-multilingual-mpnet-base-v2 sentence-transformers/multi-qa-mpnet-base-dot-v1
ALBERT	Embedding	sentence-transformers/paraphrase-albert-small-v2
Mistral	Embedding	intfloat/e5-mistral-7b-instruct Salesforce/SFR-Embedding-2_R
GTE	Embedding	Alibaba-NLP/gte-large-en-v1.5
JinaBERT	Embedding	jinaai/jina-embeddings-v2-base-en
Roberta	Sequence Classification	SamLowe/roberta-base-go_emotions

Full Changelog: huggingface/tgi-gaudi@v1.5.0...v1.5.1

v1.5.0: SynapseAI v1.17.0

SynapseAI v1.17.0

The codebase is validated with SynapseAI 1.17.0 and optimum-habana 1.13.1.