transformers
accelerate
torch==2.4.1