squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
07005806ff
flash-attention/tests/models
History
Kevin Hu 07005806ff
Add BigCode converters (#532)
2023-09-10 17:24:50 -07:00
..
test_baichuan.py [Gen] Fix calling update_graph_cache in tests 2023-09-10 17:22:37 -07:00
test_bert.py Add BigCode converters (#532) 2023-09-10 17:24:50 -07:00
test_bigcode.py Add BigCode converters (#532) 2023-09-10 17:24:50 -07:00
test_falcon.py [Gen] Fix calling update_graph_cache in tests 2023-09-10 17:22:37 -07:00
test_gpt_generation_parallel.py [Gen] Refactor decoding function 2023-09-04 17:01:38 -07:00
test_gpt_neox.py [Gen] Fix calling update_graph_cache in tests 2023-09-10 17:22:37 -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] Fix calling update_graph_cache in tests 2023-09-10 17:22:37 -07:00
test_llama.py [Gen] Fix calling update_graph_cache in tests 2023-09-10 17:22:37 -07:00
test_opt.py [Gen] Fix calling update_graph_cache in tests 2023-09-10 17:22:37 -07:00
test_vit.py Run isort and black on test files 2023-08-18 20:59:35 -07:00