ColossalAI/applications/Colossal-LLaMA/colossal_llama/utils
flybird11111 64f74a157e
[NPU]support npu (#6089)
* support npu

* support pretrain

support pretrain

fix

* support lora

fix

fix

* support chatglm

fix

fxi

fix

[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

fix

fix

[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

fix

[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

fix

fix

fix

* Update train.py

* Update train.py
2024-11-20 15:28:35 +08:00
..
__init__.py [Feature] Support LLaMA-3 CPT and ST (#5619) 2024-04-23 13:54:05 +08:00
ckpt_io.py [NPU]support npu (#6089) 2024-11-20 15:28:35 +08:00
froze.py [Feature] Support LLaMA-3 CPT and ST (#5619) 2024-04-23 13:54:05 +08:00
neftune_patch.py [Feature] Support LLaMA-3 CPT and ST (#5619) 2024-04-23 13:54:05 +08:00
stream_chat_patch.py [Feature] Support LLaMA-3 CPT and ST (#5619) 2024-04-23 13:54:05 +08:00
utils.py [Colossal-LLaMA] Refactor latest APIs (#6030) 2024-08-28 17:01:58 +08:00