requirements.txt 226 Bytes
Newer Older
1
transformers==4.19.0
2
3
4
datasets==1.16.0
wandb==0.12.0
tensorboard==2.6.0
5
6
torch==1.11.0
huggingface-hub==0.1.0
7
8
9
git+https://github.com/huggingface/accelerate.git@3c45b6f760ad8745be9ebc9bbb26f5b04dea4abe
datasketch==1.5.7
dpu_utils