accelerate>=0.16.0
torchvision
datasets
diffusers
numpy
gradio