nyu-mll/multi_nli
Viewer • Updated • 412k • 42k • 122
How to use matekadlicsko/distilroberta-nli with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("zero-shot-classification", model="matekadlicsko/distilroberta-nli") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("matekadlicsko/distilroberta-nli")
model = AutoModelForSequenceClassification.from_pretrained("matekadlicsko/distilroberta-nli", device_map="auto")This model can be used for Natural Language Inference (NLI) tasks. It is a version of roberta-base fine-tuned on multi_nli and english xnli.
The model's performance on NLI tasks is as follows: