github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/examples/images/vit/requirements.txt

6 lines
92 B
Plaintext
Raw Blame History

colossalai >= 0.1.12
torch >= 1.8.1
numpy>=1.24.1
tqdm>=4.61.2
transformers>=4.20.0
datasets
Reference in New Issue View Git Blame Copy Permalink