torch==2.7.0+cu118
transformers
accelerate