search

alexzhang13 / flashattention2-custom-mask

Triton implementation of FlashAttention2 that adds Custom Masks.
Apache License 2.0
72 stars 6 forks source link

pytorch 12.4 #8

Open alita-moore opened 2 months ago

alita-moore commented 2 months ago

Is it possible to use pytorch 12.4 with this module?