Spaces:
Build error
Build error
change jax installation to enable GPU
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
numpy>=1.16.4
|
| 2 |
-
jax>=0.2.6
|
|
|
|
| 3 |
jaxlib>=0.1.57
|
| 4 |
flax>=0.2.2
|
| 5 |
#opencv-python>=4.4.0
|
|
@@ -11,4 +12,4 @@ tensorflow-hub>=0.11.0
|
|
| 11 |
transformers==4.8.2
|
| 12 |
tqdm==4.61.2
|
| 13 |
streamlit==0.84.1
|
| 14 |
-
gdown==3.13.0
|
|
|
|
| 1 |
numpy>=1.16.4
|
| 2 |
+
#jax>=0.2.6
|
| 3 |
+
jax[cuda111]
|
| 4 |
jaxlib>=0.1.57
|
| 5 |
flax>=0.2.2
|
| 6 |
#opencv-python>=4.4.0
|
|
|
|
| 12 |
transformers==4.8.2
|
| 13 |
tqdm==4.61.2
|
| 14 |
streamlit==0.84.1
|
| 15 |
+
gdown==3.13.0
|