tmm1's picture
Merge pull request #336 from tmm1/flash-attn
0d2e34f unverified