datasets==2.11.0
pytorch-lightning[strategies]==2.0.2
tensorboardX==2.6
torch==2.0.0
transformers==4.28.1