accelerate
torch==2.4.1
torchaudio==2.5.1+cxx11.abi