Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
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
|