search

cdpierse / transformers-interpret

Model explainability that works seamlessly with 🤗 transformers. Explain your transformers model in just 2 lines of code.
Apache License 2.0
1.29k stars 97 forks source link

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

Open anonimoustt opened 5 months ago

anonimoustt commented 5 months ago

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''??