Pipeline with a fine-tuned pre-trained model
I have a pre-trained model from ‘facebook/bart-large-mnli’ I used the Trainer in order to train it on my own dataset.model = BartForSequenceClassification.from_pretrained("facebook/bart-large-mnli", num_labels=14, ignore_mismatched_sizes=True)And then after I train it, I try to use the following:# Import the Transformers pipeline library
from transformers import pipeline

# Initializing Zero-Shot Classifier
classifier = pipeline("zero-shot-classification", model=model, tokenizer=tokenizer, id2label=id2label)I get the following error from it:Failed to determine 'entailment' label id from the label2id mapping in the model config. Setting to -1. Define a descriptive label2id mapping in the model config to ensure correct outputs.I tried searching the web for a solution but I cant find anything