microsoft/ms_marco
Viewer • Updated • 1.11M • 32.3k • 256
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")