transformers
torch
diffusers
datasets[audio]
accelerate