def METHOD_NAME(q, k, v, key_padding_mask, attn_mask, static_kv):
    original_mha(
        query=q,
        key=k,
        value=v,
        key_padding_mask=key_padding_mask,
        attn_mask=attn_mask,
        static_kv=static_kv,
    )