squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
4c91621a5e
flash-attention/tests/models
History
Kevin Hu 4c91621a5e
Inverse state dict for BERT (#527)
2023-09-09 01:44:21 -07:00
..
test_baichuan.py [Gen] Use flash_attn_with_kvcache in generation 2023-09-07 08:24:43 -07:00
test_bert.py Inverse state dict for BERT (#527) 2023-09-09 01:44:21 -07:00
test_falcon.py [Gen] Refactor decoding function 2023-09-04 17:01:38 -07:00
test_gpt_generation_parallel.py [Gen] Refactor decoding function 2023-09-04 17:01:38 -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 [Gen] Use flash_attn_with_kvcache in generation 2023-09-07 08:24:43 -07:00
test_gptj.py [Gen] Refactor decoding function 2023-09-04 17:01:38 -07:00
test_llama.py [Gen] Refactor decoding function 2023-09-04 17:01:38 -07:00
test_opt.py [Gen] Refactor decoding function 2023-09-04 17:01:38 -07:00
test_vit.py Run isort and black on test files 2023-08-18 20:59:35 -07:00