ColossalAI/examples/tutorial/auto_parallel/requirements.txt

8 lines
78 B
Plaintext

torch==1.12.1
colossalai
titans
pulp
datasets
matplotlib
transformers==4.22.1