squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
96d10f6545
flash-attention/flash_attn/models
History
Tri Dao 96d10f6545 Implement LLaMa
2023-04-18 21:51:35 -07:00
..
__init__.py Add __init__.py files to subdirectories for installation 2022-11-17 16:55:44 -08:00
bert.py Implement LLaMa 2023-04-18 21:51:35 -07:00
gpt_neox.py Implement GPT-NeoX 2023-03-29 01:21:25 -07:00
gpt.py Implement LLaMa 2023-04-18 21:51:35 -07:00
gptj.py Implement LLaMa 2023-04-18 21:51:35 -07:00
llama.py Implement LLaMa 2023-04-18 21:51:35 -07:00
opt.py Implement LLaMa 2023-04-18 21:51:35 -07:00
vit.py [FusedDense] Support relu, rename FusedDenseGeluDense -> FusedMLP 2023-01-17 18:12:27 -08:00