flash-attention/flash_attn/models
2023-09-10 17:24:50 -07:00
..
__init__.py Add __init__.py files to subdirectories for installation 2022-11-17 16:55:44 -08:00
baichuan.py FEAT: add codes which supporting for baichuan-inc/Baichuan-7B (#425) 2023-08-21 11:05:06 -07:00
bert.py Add BigCode converters (#532) 2023-09-10 17:24:50 -07:00
bigcode.py Add BigCode converters (#532) 2023-09-10 17:24:50 -07:00
falcon.py Run isort and black on python files 2023-08-18 14:22:11 -07:00
gpt_neox.py [Rotary] Implement rotary in Triton 2023-09-03 02:51:58 -07:00
gpt.py Add BigCode converters (#532) 2023-09-10 17:24:50 -07:00
gptj.py Run isort and black on python files 2023-08-18 14:22:11 -07:00
llama.py Support LLaMa2 and CodeLLaMa (#491) 2023-08-30 10:31:14 -07:00
opt.py Run isort and black on python files 2023-08-18 14:22:11 -07:00
vit.py Run isort and black on python files 2023-08-18 14:22:11 -07:00