squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
a8c35b4f57
flash-attention/tests
History
GAOXinyu a8c35b4f57
FEAT: add codes which supporting for baichuan-inc/Baichuan-7B (#425)
2023-08-21 11:05:06 -07:00
..
layers Run isort and black on test files 2023-08-18 20:59:35 -07:00
losses Run isort and black on test files 2023-08-18 20:59:35 -07:00
models FEAT: add codes which supporting for baichuan-inc/Baichuan-7B (#425) 2023-08-21 11:05:06 -07:00
modules Run isort and black on test files 2023-08-18 20:59:35 -07:00
ops Run isort and black on test files 2023-08-18 20:59:35 -07:00
test_flash_attn.py Run isort and black on test files 2023-08-18 20:59:35 -07:00
test_rotary.py Run isort and black on test files 2023-08-18 20:59:35 -07:00