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