numpy<2
pytorch-lightning
torch
torchaudio
stable-audio-tools