torch
torchaudio
stable-audio-tools
einops
stable_audio_tools