requirements.txt

uvicorn==0.17.1
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
orbax==0.1.9