2023-08-15 01:27:26 +08:00
|
|
|
__version__ = "2.0.6.post2"
|
2023-07-17 20:26:11 +08:00
|
|
|
|
|
|
|
|
from flash_attn.flash_attn_interface import flash_attn_func
|
|
|
|
|
from flash_attn.flash_attn_interface import flash_attn_kvpacked_func
|
|
|
|
|
from flash_attn.flash_attn_interface import flash_attn_qkvpacked_func
|
|
|
|
|
from flash_attn.flash_attn_interface import flash_attn_varlen_qkvpacked_func
|
|
|
|
|
from flash_attn.flash_attn_interface import flash_attn_varlen_kvpacked_func
|
|
|
|
|
from flash_attn.flash_attn_interface import flash_attn_varlen_func
|
