transformers
accelerate
torch==2.6.0+cu118