Attention Map Visualization Script in "Vision Transformers Need Registers" Paper

[JihwanEom]

Hi there, thanks for merging #282!

I was wondering if you could release the script you used for visualizing the attention maps in the VISION TRANSFORMERS NEED REGISTERS paper. It caught my eye, and I'm keen on diving into it a bit more.

Appreciate it!