Spaces:
Runtime error
Runtime error
Commit
·
4ee3632
1
Parent(s):
d4f36e4
update
Browse files- requirements-cpu.txt +1 -1
- requirements.txt +2 -1
requirements-cpu.txt
CHANGED
|
@@ -2,7 +2,7 @@ einops
|
|
| 2 |
flax
|
| 3 |
gradio
|
| 4 |
huggingface-hub
|
| 5 |
-
|
| 6 |
jaxlib
|
| 7 |
ml_collections
|
| 8 |
numpy
|
|
|
|
| 2 |
flax
|
| 3 |
gradio
|
| 4 |
huggingface-hub
|
| 5 |
+
jax
|
| 6 |
jaxlib
|
| 7 |
ml_collections
|
| 8 |
numpy
|
requirements.txt
CHANGED
|
@@ -3,7 +3,8 @@ flax
|
|
| 3 |
gradio
|
| 4 |
huggingface-hub
|
| 5 |
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
| 6 |
-
jax[cuda12_pip]~=0.4.25
|
|
|
|
| 7 |
jaxlib
|
| 8 |
ml_collections
|
| 9 |
numpy
|
|
|
|
| 3 |
gradio
|
| 4 |
huggingface-hub
|
| 5 |
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
| 6 |
+
#jax[cuda12_pip]~=0.4.25
|
| 7 |
+
jax[cuda12_pip]==0.4.25
|
| 8 |
jaxlib
|
| 9 |
ml_collections
|
| 10 |
numpy
|