requirements.txt

datasets
torch
torchvision
git+https://github.com/huggingface/diffusers.git@30f6f44
transformers
accelerate
ftfy
xformers
modin[all]