Visualizing Attention Maps in Pre-trained Vision Transformers (Pytorch)

Goal: Visualizing the attention maps for the CLS token in a pretrained Vision Transformer from the timm library.