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-L4 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-L4 with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("cross-encoder/ms-marco-TinyBERT-L4") 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-L4 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("cross-encoder/ms-marco-TinyBERT-L4") model = AutoModelForSequenceClassification.from_pretrained("cross-encoder/ms-marco-TinyBERT-L4", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Fix training code link
Verified signature77fdad7
-
Revert inadvertent config, tokenizer updates
Tom AarsenThis reverts commit b989367cd22a5bbf9102f4f098e793c27d314346.8b65c37
-
Add exported openvino model 'openvino_model_qint8_quantized.xml'
Verified signature87f7576 -
Add new CrossEncoder model
Verified signature73f47cd -
Add exported onnx model 'model_qint8_avx512_vnni.onnx'
Verified signature4847800 -
Add exported onnx model 'model_qint8_avx512.onnx'
Verified signaturea7b6212 -
Add exported onnx model 'model_quint8_avx2.onnx'
Verified signaturebb61e3a -
Add exported onnx model 'model_qint8_arm64.onnx'
Verified signaturee8f03ba -
Add exported onnx model 'model_O4.onnx'
Verified signature6ae762f -
Add exported onnx model 'model_O3.onnx'
Verified signature2aeb0f0 -
Add exported onnx model 'model_O2.onnx'
Verified signature2809984 -
Add exported onnx model 'model_O1.onnx'
Verified signature05b746f -
Add new CrossEncoder model
Verified signatureb989367
-
Push tokenizer again (#2)
Verified signature- Push tokenizer again (99d21b7c4a024a9aa280c4b649caea20920521d5)41e7f1b
-
Update model metadata
Verified signature5942357
-
L-\d -> L\d
Verified signature8efadfc
-
Adding `safetensors` variant of this model (#1)
Verified signature- Adding `safetensors` variant of this model (a48e8c8d3a486460437e93878fef493b05cc723f) Co-authored-by: Safetensors convertbot <SFconvertbot@users.noreply.huggingface.co>f960fa5
-
Update README.md
12a9f22
-
upload flax model
0bb112e -
allow flax
18a6b5f
-
up
nreimers6354f73
-
upload
nreimerse6d00b4 -
initial commit
f7ae382