codeparrot-small / requirements.txt
lvwerra's picture
lvwerra HF staff
Create requirements.txt
f3ed84c
torch==1.9.0
wandb
tensorboard
transformers==4.12.2
datasets==1.13.0
accelerate==0.5.1