Update requirements.txt

requirements.txt

@@ -1,5 +1,17 @@
-orbax-checkpoint==0.2.4
-uvicorn==0.10.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
+pydantic==1.*
 
+gdown==4.6.4
+pytorch_lightning
+ftfy
+OmegaConf
+orbax==0.1.2