Camille Zhong 44ca61a22b [llama] fix neftune & pbar with start_step (#5364)		2024-02-05 18:04:23 +08:00
..
__init__.py	initial commit: add colossal llama 2 (#4784)	2023-09-24 23:12:26 +08:00
ckpt_io.py	initial commit: add colossal llama 2 (#4784)	2023-09-24 23:12:26 +08:00
flash_attention_patch.py	[llama] add flash attn patch for npu (#5362)	2024-02-05 16:48:34 +08:00
froze.py	initial commit: add colossal llama 2 (#4784)	2023-09-24 23:12:26 +08:00
neftune_patch.py	[llama] fix neftune & pbar with start_step (#5364)	2024-02-05 18:04:23 +08:00