Text Ranking
Transformers
ONNX
Safetensors
Transformers.js
sentence-transformers
English
deberta-v2
text-classification
reranker
text-embeddings-inference
Instructions to use mixedbread-ai/mxbai-rerank-base-v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mixedbread-ai/mxbai-rerank-base-v1 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("mixedbread-ai/mxbai-rerank-base-v1") model = AutoModelForSequenceClassification.from_pretrained("mixedbread-ai/mxbai-rerank-base-v1", device_map="auto") - Transformers.js
How to use mixedbread-ai/mxbai-rerank-base-v1 with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('text-ranking', 'mixedbread-ai/mxbai-rerank-base-v1'); - sentence-transformers
How to use mixedbread-ai/mxbai-rerank-base-v1 with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("mixedbread-ai/mxbai-rerank-base-v1") query = "Which planet is known as the Red Planet?" passages = [ "Venus is often called Earth's twin because of its similar size and proximity.", "Mars, known for its reddish appearance, is often referred to as the Red Planet.", "Jupiter, the largest planet in our solar system, has a prominent red spot.", "Saturn, famous for its rings, is sometimes mistaken for the Red Planet." ] scores = model.predict([(query, passage) for passage in passages]) print(scores) - Notebooks
- Google Colab
- Kaggle
Commit History
Update README.md 1e7337d verified
Update LICENSE 9ed7b04 verified
Update README.md e1b01fd verified
add citation ff16ff8 verified
Update README.md 03241da verified
Add LICENSE ca0d2db verified
Update tokenizer_config.json 94825b6 verified
Update README.md 1197752 verified
Update README.md (#2) 467c15a verified
Update README.md bb6017b verified
Update README.md 67fea40 verified
Update README.md 3e71073 verified
Update README.md 1152fd6 verified
Update README.md 49e4339 verified
Aamir commited on
Update README.md 01d507b verified
Aamir commited on
Upload tokenizer 3ee5f37 verified
Aamir commited on
Upload DebertaV2ForSequenceClassification 29d241e verified
Aamir commited on
initial commit f54905c verified
Aamir commited on