Почетна Преглед Помоћ
Регистрација Пријавите се
david
/
flash-attention
огледало од https://github.com/Dao-AILab/flash-attention
1
0
Креирај огранак 0
Датотеке Дискусије 0 Вики
Грана: main
Гране Ознаке
flash-attention
/
flash_attn
/
modules
SueJane 3f1b4d38e7 Fix: check the type of max_seqlen_k instead of checking max_seqlen twice (#1127) пре 4 месеци
..
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation пре 2 година
block.py abbc131173 [LayerNorm] Switch from CUDA to Triton implementation пре 11 месеци
embedding.py f1a73d0740 Run isort and black on python files пре 1 година
mha.py 3f1b4d38e7 Fix: check the type of max_seqlen_k instead of checking max_seqlen twice (#1127) пре 4 месеци
mlp.py c3b2196652 Add Alibi to MHA, test with Baichuan-13B пре 11 месеци