microsoft/ms_marco
Viewer โข Updated โข 1.11M โข 21.2k โข 263
How to use naver/trecdl22-crossencoder-roberta with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="naver/trecdl22-crossencoder-roberta") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("naver/trecdl22-crossencoder-roberta")
model = AutoModelForSequenceClassification.from_pretrained("naver/trecdl22-crossencoder-roberta", device_map="auto")