Fixes the inconsistency of the optionality of attention_mask #37153
Zephyr271828
changed the title Yufeng xu Fixes the inconsistency of the optionality of attention_mask 1 year ago
Zephyr271828
marked this pull request as ready for review 1 year ago
debugging issue 36758
b16084e7
debugging issue 36758
043c2465
debugging issue 36758
d4757b1d
updated attn_mask type specification in _flash_attention_forward
7d234f44
removed pdb
e66e8275
added a blank line
585ce298
removed indentation
1cbd59df
Merge branch 'main' into Yufeng-Xu
516836ea
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub