torch==1.9.0
wandb
tensorboard
transformers==4.12.2
datasets==1.13.0
accelerate==0.5.1