requirements.txt

jax[cpu]
flax
git+https://github.com/huggingface/diffusers
accelerate
transformers
ftfy
OmegaConf
huggingface-hub==0.11.0
torch
safetensors
gdown
requests