32a953f486
* Bump version to 2.0.2 Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com> * Update version in Dockerfile Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com> --------- Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
9 lines
442 B
Python
9 lines
442 B
Python
__version__ = "2.0.2"
|
|
|
|
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
|