ColossalAI/examples/language/gpt/experiments/pipeline_parallel/requirements.txt

3 lines
36 B
Plaintext

colossalai >= 0.1.12
torch >= 1.8.1