src/transformers/models/blip_2/modeling_blip_2.py/Blip2QFormerMultiHeadAttention/get_attention_map
class Blip2QFormerMultiHeadAttention:    def get_attention_map(self):
        return self.attention_map