Página inicial Explorar Ajuda
Registrar Entrar
david
/
flash-attention
mirror de https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Arquivos Issues 0 Wiki
Branch: main
Branches Tags
flash-attention
/
flash_attn
/
layers
Tri Dao 8c20cfef49 [Rotary] Support qkv block layout from GQA 3 meses atrás
..
__init__.py ece539abd6 Add __init__.py files to subdirectories for installation 2 anos atrás
patch_embed.py f1a73d0740 Run isort and black on python files 1 ano atrás
rotary.py 8c20cfef49 [Rotary] Support qkv block layout from GQA 3 meses atrás