requirements.txt

jax==0.4.6
jaxlib==0.4.6
flax==0.6.6
diffusers==0.13.1
accelerate==0.16.0
transformers==4.26.1
huggingface-hub==0.25.2
torch==1.13.1
safetensors==0.3.1
pydantic==1.*

gdown==4.6.4
pytorch_lightning
ftfy
OmegaConf
orbax==0.1.2