ColossalAI/examples/language/llama/requirements.txt
binmakeswell f4c5aafe29
[example] llama3 (#5631)
* release llama3

* [release] llama3

* [release] llama3

* [release] llama3

* [release] llama3
2024-04-23 18:48:07 +08:00

9 lines
111 B
Plaintext

colossalai>=0.3.6
datasets
numpy
tqdm
transformers
flash-attn>=2.0.0
SentencePiece==0.1.99
tensorboard==2.14.0