Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
jax~=0.4.13
|
|
|
2 |
flax~=0.7.1
|
3 |
fjutils~=0.0.15
|
4 |
numpy==1.25.2
|
@@ -12,4 +13,5 @@ tqdm==4.65.0
|
|
12 |
datasets==2.14.3
|
13 |
setuptools~=60.0.0
|
14 |
easydel
|
15 |
-
wandb
|
|
|
|
1 |
jax~=0.4.13
|
2 |
+
jaxlib~=0.4.13
|
3 |
flax~=0.7.1
|
4 |
fjutils~=0.0.15
|
5 |
numpy==1.25.2
|
|
|
13 |
datasets==2.14.3
|
14 |
setuptools~=60.0.0
|
15 |
easydel
|
16 |
+
wandb
|
17 |
+
torch
|