grouped-query-attention-pytorch
grouped-query-attention-pytorch copied to clipboard
function param support attn_mask ?
Can i open the comments here? i want use this func for attn_mask