requirements.txt

pip==23.2.1
uvicorn==0.17.1
orbax==0.1.9
pydantic==1.*
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