transformers
accelerate
torch==2.6.0+cu118
torchaudio==2.4.1