Attention map extraction

[yi-ming-qian]

Hello, thanks for sharing the work, it is very inspiring. I wonder if you can share the attention extraction and visualization script used for creating Figure 2 in the paper?

[zzxslp]

Hi! We use the code in this repo. Currently swamped with other things but can share the probing code for LLaVA later.