torchmetrics==0.7.0
torchsynth==1.0.2
soundfile
einops
pytorch-ssim
piqa
torchinfo
mido
tensorboard
librosa
transformers