requirements.txt

jax[cpu]==0.4.4
flax==0.6.6
diffusers==0.13.1
accelerate==0.16.0
transformers==4.26.1
huggingface-hub==0.11.0
torch==1.13.1
safetensors==0.2.8
gdown==4.6.4
pytorch_lightning
ftfy
OmegaConf