accelerate
torch==2.4.0
transformers