requirements.txt 118 Bytes
Newer Older
yangzhong's avatar
yangzhong committed
1
2
3
4
5
6
7
numpy>=1.21.0
psutil>=5.8.0
gpustat>=1.0.0
torch>=2.0.0
transformers>=4.30.0
datasets>=2.10.0
huggingface_hub>=0.16.0