Closed brewswang closed 1 year ago
@brewswang Please consider renaming the function based on flash-attn2, they changed the function names.
thanks,
These functions have been renamed:
flash_attn_unpadded_func
-> flash_attn_varlen_func
flash_attn_unpadded_qkvpacked_func
-> flash_attn_varlen_qkvpacked_func
flash_attn_unpadded_kvpacked_func
-> flash_attn_varlen_kvpacked_func
希望对其他人有帮助
ImportErrorImportError: : cannot import name 'flash_attn_unpadded_qkvpacked_func'