ColossalAI/examples/language/grok-1/requirements.txt

4 lines
59 B
Plaintext

torch>=2.1.0,<2.2.0
colossalai>=0.3.6
transformers==4.35.0