 Zhihao Shen
|
30e1ef0f79
minify torch.torch.int32 to torch.int32 (#1237)
|
il y a 2 mois |
 Ying Zhang
|
8cbc8a042f
small fixes
|
il y a 2 mois |
|
Ying Zhang
|
cdbbe844b1
minor changes to unpad_input test util func
|
il y a 3 mois |
|
Ying Zhang
|
db80387343
Add seqused_q in fwd / bwd and seqused_k in bwd.
|
il y a 3 mois |
 Avelina9X
|
c94cd09744
Updated missing docstrings for args and returns in bert_padding.py (#795)
|
il y a 10 mois |
 Su Zhu
|
8f6f48d8a8
add unpad_input_for_concatenated_sequences (#499)
|
il y a 1 an |
 Tri Dao
|
f1a73d0740
Run isort and black on python files
|
il y a 1 an |
 Antoine Adam
|
4e38df059e
remove numpy dependency
|
il y a 2 ans |
|
Tri Dao
|
6cc7342575
Support index_first_axis with more than 2 dimensions
|
il y a 2 ans |
|
Tri Dao
|
5a61cb7729
Rename src -> flash_attn
|
il y a 2 ans |
