accelerate
scipy
diffusers[torch]
safetensors
torch==2.0.1+cu117
torchvision
torchmetrics
omegaconf
transformers