cdpierse / transformers-interpret

Model explainability that works seamlessly with 🤗 transformers. Explain your transformers model in just 2 lines of code.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error :AssertionError: Forward hook did not obtain any outputs for given layer

anonimoustt opened this issue · comments

I was trying following code and found an error

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"],
)
AssertionError: Forward hook did not obtain any outputs for given layer

Limitation: So your library cannot generate explanations for the outcomes generated by ''facebook/bart-large-mnli''??