src/transformers/models/blip_2/modeling_blip_2.py/Blip2QFormerMultiHeadAttention/save_attention_map
class Blip2QFormerMultiHeadAttention:    def save_attention_map(self, attention_map):
        self.attention_map = attention_map