david/flash-attention

mirror of https://github.com/Dao-AILab/flash-attention

Author	SHA1 Message	Date
Tri Dao	3f7d5786ba Pass alibi slopes to flash_attn_with_kvcache during generation	11 months ago
Tri Dao	2c7d7b7396 Implement norm head for Baichuan2	11 months ago
Tri Dao	c3b2196652 Add Alibi to MHA, test with Baichuan-13B	1 year ago
GAOXinyu	a8c35b4f57 FEAT: add codes which supporting for baichuan-inc/Baichuan-7B (#425)	1 year ago