ColossalAI/examples/language/opt/requirements.txt

7 lines
98 B
Plaintext

colossalai
torch >= 1.8.1
datasets >= 1.8.0
sentencepiece != 0.1.92
protobuf
accelerate == 0.13.2