NickKolok commited on
Commit
92a2ff9
1 Parent(s): 44fd277

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +14 -2
requirements.txt CHANGED
@@ -1,5 +1,17 @@
1
- orbax-checkpoint==0.2.4
2
- uvicorn==0.10.1
 
 
 
 
 
 
 
3
 
 
 
 
 
 
4
 
5
 
 
1
+ jax[cpu]>=0.4.4
2
+ flax==0.6.6
3
+ diffusers>=0.13.1
4
+ accelerate==0.16.0
5
+ transformers==4.26.1
6
+ huggingface-hub>=0.15.0
7
+ torch==1.13.1
8
+ safetensors==0.2.8
9
+ pydantic==1.*
10
 
11
+ gdown==4.6.4
12
+ pytorch_lightning
13
+ ftfy
14
+ OmegaConf
15
+ orbax==0.1.2
16
 
17