File tree Expand file tree Collapse file tree 4 files changed +9
-212
lines changed Expand file tree Collapse file tree 4 files changed +9
-212
lines changed Load Diff This file was deleted.
Load Diff This file was deleted.
Original file line number Diff line number Diff line change 11__version__ = "2.8.3"
22
3- # NOTE(timmy): Comment out these so that we can run cute version without unneeded dependencies
4- # from flash_attn.flash_attn_interface import (
5- # flash_attn_func,
6- # flash_attn_kvpacked_func,
7- # flash_attn_qkvpacked_func,
8- # flash_attn_varlen_func,
9- # flash_attn_varlen_kvpacked_func,
10- # flash_attn_varlen_qkvpacked_func,
11- # flash_attn_with_kvcache,
12- # )
3+ from flash_attn .flash_attn_interface import (
4+ flash_attn_func ,
5+ flash_attn_kvpacked_func ,
6+ flash_attn_qkvpacked_func ,
7+ flash_attn_varlen_func ,
8+ flash_attn_varlen_kvpacked_func ,
9+ flash_attn_varlen_qkvpacked_func ,
10+ flash_attn_with_kvcache ,
11+ )
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments