Spaces:
Runtime error
Runtime error
Commit
•
63465b9
1
Parent(s):
4ee3632
update
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -2,9 +2,9 @@ einops
|
|
| 2 |
flax
|
| 3 |
gradio
|
| 4 |
huggingface-hub
|
| 5 |
-
|
| 6 |
#jax[cuda12_pip]~=0.4.25
|
| 7 |
-
jax[
|
| 8 |
jaxlib
|
| 9 |
ml_collections
|
| 10 |
numpy
|
|
|
|
| 2 |
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 |
+
jax[cuda12]
|
| 8 |
jaxlib
|
| 9 |
ml_collections
|
| 10 |
numpy
|