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-L6 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-L6 with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("cross-encoder/ms-marco-TinyBERT-L6") 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-L6 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("cross-encoder/ms-marco-TinyBERT-L6") model = AutoModelForSequenceClassification.from_pretrained("cross-encoder/ms-marco-TinyBERT-L6") - Notebooks
- Google Colab
- Kaggle
Commit History
Revert inadvertent config, tokenizer updates 7e02a11
Tom Aarsen commited on
Add exported openvino model 'openvino_model_qint8_quantized.xml' 65a498f verified
Add new CrossEncoder model c12aee1 verified
Add exported onnx model 'model_qint8_avx512_vnni.onnx' 2a505d3 verified
Add exported onnx model 'model_qint8_avx512.onnx' 06f17ad verified
Add exported onnx model 'model_quint8_avx2.onnx' e6ffac4 verified
Add exported onnx model 'model_qint8_arm64.onnx' 6ba0eb8 verified
Add exported onnx model 'model_O4.onnx' 73bf0a4 verified
Add exported onnx model 'model_O3.onnx' 08ac1e1 verified
Add exported onnx model 'model_O2.onnx' 3a83364 verified
Add exported onnx model 'model_O1.onnx' b123694 verified
Add new CrossEncoder model 8dd6d11 verified
Update model metadata 57d25c6 verified
L-\d -> L\d 946a29b verified
Adding `safetensors` variant of this model (#1) 48d154f verified
Update README.md 36aded8
upload flax model 9e03713
allow flax ca11e8a
up cb613d5
nreimers commited on
upload 67d6b31
nreimers commited on