accelerate==0.18.0 datasets==2.11.0 numpy==1.24.2 torch==2.0.0 transformers==4.27.4