ColossalAI/examples/language/opt/requirements.txt

5 lines
76 B
Plaintext

colossalai >= 0.3.2
torch >= 1.8.1
datasets >= 1.8.0
transformers >= 4.30.2