Tri Dao a86442f0f3 [Gen] Use flash_attn_with_kvcache in generation		2023-09-07 08:24:43 -07:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
benchmark.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
distributed.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
generation.py	[Gen] Use flash_attn_with_kvcache in generation	2023-09-07 08:24:43 -07:00
pretrained.py	[GPT] Fix loading weights from HF hub	2023-08-21 22:56:02 -07:00