src/transformers/models/blip/modeling_blip_text.py/BlipTextSelfAttention/save_attention_map
class BlipTextSelfAttention:    def save_attention_map(self, attention_map):
        self.attention_map = attention_map