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