torch==2.7.1
torchaudio==2.4.1
transformers
accelerate