torch==2.5.1
transformers
accelerate