Skip to content

Releases: dptech-corp/flash-attention

odd-len-support

29 Jan 09:47
2c5d881

Choose a tag to compare

Add odd length support (#5)

* add odd length support

* add mask in attn_mask & attn_bias

* rm useless files

* move if to the outer loop

* remove comments

---------

Co-authored-by: xhj <jixh@dp.tech>

Attention mask & Attention bias support

13 Oct 03:16

Choose a tag to compare

refs/heads/workflow

add .