Text Ranking
sentence-transformers
PyTorch
JAX
ONNX
Safetensors
OpenVINO
Transformers
English
bert
text-classification
text-embeddings-inference
Instructions to use cross-encoder/ms-marco-TinyBERT-L2-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use cross-encoder/ms-marco-TinyBERT-L2-v2 with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("cross-encoder/ms-marco-TinyBERT-L2-v2") 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) - Transformers
How to use cross-encoder/ms-marco-TinyBERT-L2-v2 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("cross-encoder/ms-marco-TinyBERT-L2-v2") model = AutoModelForSequenceClassification.from_pretrained("cross-encoder/ms-marco-TinyBERT-L2-v2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
Revert inadvertent config, tokenizer updates d327be6
Tom Aarsen commited on
Add exported openvino model 'openvino_model_qint8_quantized.xml' c4982bc verified
Add new CrossEncoder model aab7c05 verified
Add exported onnx model 'model_qint8_avx512_vnni.onnx' 06bb80f verified
Add exported onnx model 'model_qint8_avx512.onnx' 87f54cd verified
Add exported onnx model 'model_quint8_avx2.onnx' 4ab9bb9 verified
Add exported onnx model 'model_qint8_arm64.onnx' 1c612c6 verified
Add exported onnx model 'model_O4.onnx' 4305d3b verified
Add exported onnx model 'model_O3.onnx' aeeec00 verified
Add exported onnx model 'model_O2.onnx' c77c985 verified
Add exported onnx model 'model_O1.onnx' e7ec13c verified
Add new CrossEncoder model 15192f1 verified
Update model metadata ad21154 verified
L-\d -> L\d 75b56ed verified
Adding `safetensors` variant of this model (#4) 841d331 verified
Update README.md e9ea268
upload flax model 4a7941f
allow flax 06a3f0e
update config 2673f40
nreimers commited on
upload 643fe93
nreimers commited on