squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
3557e0bb8f
flash-attention/tests/models
History
Tri Dao 798858f9f1 Fix test_baichuan
2023-09-03 21:01:37 -07:00
..
test_baichuan.py Fix test_baichuan 2023-09-03 21:01:37 -07:00
test_bert.py [GPT] Fix loading weights from HF hub 2023-08-21 22:56:02 -07:00
test_falcon.py Fix test_baichuan 2023-09-03 21:01:37 -07:00
test_gpt_generation_parallel.py [Rotary] Implement rotary in Triton 2023-09-03 02:51:58 -07:00
test_gpt_neox.py Run isort and black on test files 2023-08-18 20:59:35 -07:00
test_gpt_parallel.py Run isort and black on test files 2023-08-18 20:59:35 -07:00
test_gpt.py Support MQA + MP for decoding (#490) 2023-08-30 10:29:54 -07:00
test_gptj.py Run isort and black on test files 2023-08-18 20:59:35 -07:00
test_llama.py Run isort and black on test files 2023-08-18 20:59:35 -07:00
test_opt.py [GPT] Move GPT and OPT generation tests to test_{gpt,opt}.py 2023-08-26 12:55:02 -07:00
test_vit.py Run isort and black on test files 2023-08-18 20:59:35 -07:00