How to use typeform/distilbert-base-uncased-mnli with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("typeform/distilbert-base-uncased-mnli") model = AutoModelForSequenceClassification.from_pretrained("typeform/distilbert-base-uncased-mnli", device_map="auto")
hhschu +1: Ezi - Model Card (c213a4b99cbb618477949c1483b6916d023c2e45) Co-authored-by: Ezi Ozoani <Ezi@users.noreply.huggingface.co>996dacf
hhschu +1: joaogante - Add TF weights (20fae5e389891a08e888e902758ed97384cdf1b7) Co-authored-by: Joao Gante <joaogante@users.noreply.huggingface.co>9487bd4