accelerate==0.32.1 datasets==2.20.0 torch==2.3.1 transformers==4.43.2