github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-19 16:37:13 +00:00
Code Issues Packages Projects Releases Wiki Activity
dev/zero-offload
ColossalAI/applications/Colossal-LLaMA-2/colossal_llama2/utils
History
pre-commit-ci[bot] df612434c9 [pre-commit.ci] auto fixes from pre-commit.com hooks 
for more information, see https://pre-commit.ci
2024-06-14 16:27:46 +08:00
..
__init__.py initial commit: add colossal llama 2 (#4784) 2023-09-24 23:12:26 +08:00
ckpt_io.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-14 16:27:46 +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
stream_chat_patch.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-14 16:27:46 +08:00