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.15.0
torch==1.13.1
safetensors==0.2.8
pydantic==1.*

gdown==4.6.4
pytorch_lightning
ftfy
OmegaConf
orbax==0.1.2