How to apply this work on google/vit model from hugging face ?

Thanks for the great work,

When I tried to follow the vit notebook I got a lot of attribute errors and I think the model architecture and layers names may not be the same with ```google/vit-base-patch16-224-in21k``` so Can you please help me with the steps and modifications needed to apply the explainability on the mentioned model.