torch==2.2.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu transformers==4.40.0 datasets==2.18.0 accelerate==0.29.0