squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
ef6d8c75d9
flash-attention/flash_attn/models
History
GAOXinyu a8c35b4f57
FEAT: add codes which supporting for baichuan-inc/Baichuan-7B (#425)
2023-08-21 11:05:06 -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 Run isort and black on python files 2023-08-18 14:22:11 -07:00
falcon.py Run isort and black on python files 2023-08-18 14:22:11 -07:00
gpt_neox.py Run isort and black on python files 2023-08-18 14:22:11 -07:00
gpt.py handle uneven heads across ranks when combining state_dicts; resolves #467 (#468) 2023-08-20 14:57:34 -07:00
gptj.py Run isort and black on python files 2023-08-18 14:22:11 -07:00
llama.py map custom model state_dict back to huggingface format (#465) 2023-08-18 20:51:39 -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