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:
# pip install -U transformers accelerate # 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", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Commit History
-
Add exported onnx model 'model_qint8_avx512_vnni.onnx'
Verified signature2a505d3 -
Add exported onnx model 'model_qint8_avx512.onnx'
Verified signature06f17ad -
Add exported onnx model 'model_quint8_avx2.onnx'
Verified signaturee6ffac4 -
Add exported onnx model 'model_qint8_arm64.onnx'
Verified signature6ba0eb8 -
Add exported onnx model 'model_O4.onnx'
Verified signature73bf0a4 -
Add exported onnx model 'model_O3.onnx'
Verified signature08ac1e1 -
Add exported onnx model 'model_O2.onnx'
Verified signature3a83364 -
Add exported onnx model 'model_O1.onnx'
Verified signatureb123694 -
Add new CrossEncoder model
Verified signature8dd6d11
-
Push tokenizer again (#3)
Verified signature- Push tokenizer again (5f63b1b8c8e00f89b247c885605e0e3fbd73fca8)9522e04
-
Update model metadata
Verified signature57d25c6
-
L-\d -> L\d
Verified signature946a29b
-
Adding `safetensors` variant of this model (#1)
Verified signature- Adding `safetensors` variant of this model (de255114cced5675aaa7f95aab3773438b359999) Co-authored-by: Safetensors convertbot <SFconvertbot@users.noreply.huggingface.co>48d154f
-
Update README.md
36aded8
-
upload flax model
9e03713 -
allow flax
ca11e8a
-
up
nreimerscb613d5
-
upload
nreimers67d6b31 -
initial commit
d0a17bf