mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-02 08:16:48 +00:00
4a68efb7da
* refactor latest code * update api * add dummy dataset * update Readme * add setup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update files * add PP support * update arguments * update argument * reorg folder * update version * remove IB infor * update utils * update readme * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update save for zero * update save * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * add apex * update --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
16 lines
225 B
Plaintext
16 lines
225 B
Plaintext
torch==2.1.2
|
|
huggingface-hub
|
|
packaging==24.0
|
|
colossalai>=0.4.0
|
|
autoflake==2.2.1
|
|
black==23.9.1
|
|
transformers>=4.39.3
|
|
tensorboard==2.14.0
|
|
six==1.16.0
|
|
datasets
|
|
ninja==1.11.1
|
|
flash-attn
|
|
tqdm
|
|
sentencepiece==0.1.99
|
|
protobuf<=3.20.0
|