datasets==2.11.0
deepspeed==0.9.1
pytorch-lightning==2.0.2
torch==2.0.0
transformers==4.28.1