nyu-mll/multi_nli
Viewer • Updated • 412k • 15.3k • 114
How to use kweinmeister/distilbert-mnli with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("kweinmeister/distilbert-mnli")
model = AutoModelForSequenceClassification.from_pretrained("kweinmeister/distilbert-mnli", device_map="auto")A fine-tuned version of distilbert/distilbert-base-uncased using the nyu-mll/multi_nli dataset.