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 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 with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("cross-encoder/ms-marco-TinyBERT-L2") 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 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("cross-encoder/ms-marco-TinyBERT-L2") model = AutoModelForSequenceClassification.from_pretrained("cross-encoder/ms-marco-TinyBERT-L2", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Commit History
-
Fix training code link
Verified signaturec918718
-
Revert inadvertent config, tokenizer updates
Tom AarsenThis reverts commit 0f14cfa4c90b004e1348d68147709121ace7d795.ed629aa
-
Add exported openvino model 'openvino_model_qint8_quantized.xml'
Verified signature4af3341 -
Add new CrossEncoder model
Verified signatureb5c7e5e -
Add exported onnx model 'model_qint8_avx512_vnni.onnx'
Verified signature6c0a596 -
Add exported onnx model 'model_qint8_avx512.onnx'
Verified signature2a80079 -
Add exported onnx model 'model_quint8_avx2.onnx'
Verified signaturefd22621 -
Add exported onnx model 'model_qint8_arm64.onnx'
Verified signaturec60734f -
Add exported onnx model 'model_O4.onnx'
Verified signaturecce0729 -
Add exported onnx model 'model_O3.onnx'
Verified signaturefd66467 -
Add exported onnx model 'model_O2.onnx'
Verified signaturef61a476 -
Add exported onnx model 'model_O1.onnx'
Verified signaturea895f92 -
Add new CrossEncoder model
Verified signature0f14cfa
-
Push tokenizer again (#2)
Verified signature- Push tokenizer again (071e03ddf5970974b6e0a1a84c803e1383546a78)830b8f2
-
Add link to v2 model
Verified signature09b0ed5 -
Update model metadata
Verified signature522ffbc
-
Update README.md
Verified signatureb2509b8 -
Update README.md
Verified signaturec4379fe
-
L-\d -> L\d
Verified signature8bf1a30
-
Adding `safetensors` variant of this model (#1)
Verified signature- Adding `safetensors` variant of this model (2ae3243d443902fed5d8f8fd17366c6037c8c6d6) Co-authored-by: Safetensors convertbot <SFconvertbot@users.noreply.huggingface.co>95cc886
-
Update README.md
e9ed047
-
upload flax model
33b6f71 -
allow flax
dd98073
-
up
nreimerse81a587
-
upload
nreimersd87548d -
initial commit
8b151e5