torch
torchaudio
transformers
accelerate