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
Atzars: fa3-kvcache-gqa
Atzari Tagi
flash-attention
/
flash_attn
/
utils
Tri Dao 320fb59487 Update citation 6 mēneši atpakaļ
..
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation 2 gadi atpakaļ
benchmark.py 79bd1a2d5d [LayerNorm] Implement residual + LayerNorm/RMSNorm in Triton 1 gadu atpakaļ
distributed.py f1a73d0740 Run isort and black on python files 1 gadu atpakaļ
generation.py 320fb59487 Update citation 6 mēneši atpakaļ
pretrained.py ef6d8c75d9 [GPT] Fix loading weights from HF hub 1 gadu atpakaļ