Error using LimeSaliency

[khalilmaachou]

Hello as always thanks for the work you have done, But i have a question , using the paper's parameters with LimeSaliency there is always an error in the prediction phase, here :+1: output_dict_orig['attentions'] = output_dict_orig['attentions'] * (1 - kept_tokens).float() Link because the shape of both tensors attention is (1,615) and kept_tokens is (4,827) we can't multiply 615 with 827, Thank you.