Issue with Zero Shot Classifier

captum==0.6.0
transformers==4.28.1
transformers-interpret==0.10.0
python 3.7.4

When I run the code below, I get error `AssertionError: Forward hook did not obtain any outputs for given layer`
I have tried running it locally and in Google Colab with both having the same result

```
from transformers import AutoModelForSequenceClassification, AutoTokenizer
from transformers_interpret import ZeroShotClassificationExplainer

tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-mnli")

model = AutoModelForSequenceClassification.from_pretrained("facebook/bart-large-mnli")

zero_shot_explainer = ZeroShotClassificationExplainer(model, tokenizer)

word_attributions = zero_shot_explainer(
"Today apple released the new Macbook showing off a range of new features found in the proprietary silicon chip computer. ",
labels = ["finance", "technology", "sports"],
)