torch==2.7.1
transformers
accelerate