Individual Layer Attention #4
Description
Hey I wanted to see block by block attention map(like the upsampling and downsampling blocks) visualization of the attribution prompt instead of the final aggregated one. Is there any way to do it? Any help would be appreciated.