ColossalAI/examples/tutorial/auto_parallel/requirements.txt
2023-03-28 10:40:07 +08:00

8 lines
78 B
Plaintext

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