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