mirror of https://github.com/hpcaitech/ColossalAI
10 lines
141 B
Plaintext
10 lines
141 B
Plaintext
|
colossalai>=0.3.0
|
||
|
datasets
|
||
|
numpy
|
||
|
torch>=1.12.0,<=2.0.0
|
||
|
tqdm
|
||
|
transformers
|
||
|
flash-attn>=2.0.0,<=2.0.5
|
||
|
SentencePiece==0.1.99
|
||
|
tensorboard==2.14.0
|