torch==2.3.1
torchaudio==2.3.1
torchvision==0.18.1
transformers==4.30.2
matplotlib
scipy
jsonlines
pytorch_lightning==2.3.0
einops
diffusers
soundfile
fairscale
timm
positional_encodings
wandb
xformers==0.0.27