Update requirements.txt
Browse files- requirements.txt +14 -2
requirements.txt
CHANGED
@@ -1,5 +1,17 @@
|
|
1 |
-
|
2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
|