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