Sākums Izpētīt Palīdzība
Reģistrēties Pierakstīties
david
/
flash-attention
spogulis no https://github.com/Dao-AILab/flash-attention
1
0
Atdalīts 0
Faili Problēmas 0 Vikivietne
Koks: 4ead9bd7cc
Atzari Tagi
flash-attention
/
flash_attn
/
modules
Markus Krimmel 6bbc532388 fix: cast the alibi slopes to torch.float32 (#846) 10 mēneši atpakaļ
..
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation 2 gadi atpakaļ
block.py abbc131173 [LayerNorm] Switch from CUDA to Triton implementation 1 gadu atpakaļ
embedding.py f1a73d0740 Run isort and black on python files 1 gadu atpakaļ
mha.py 6bbc532388 fix: cast the alibi slopes to torch.float32 (#846) 10 mēneši atpakaļ
mlp.py c3b2196652 Add Alibi to MHA, test with Baichuan-13B 1 gadu atpakaļ