Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update requirements.txt
Browse files- requirements.txt +1 -2
requirements.txt
CHANGED
@@ -13,8 +13,7 @@ dill
|
|
13 |
flax
|
14 |
--pre
|
15 |
--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
16 |
-
jax[
|
17 |
-
jaxlib
|
18 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
19 |
torch
|
20 |
torchaudio
|
|
|
13 |
flax
|
14 |
--pre
|
15 |
--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
16 |
+
jax[cuda_local]
|
|
|
17 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
18 |
torch
|
19 |
torchaudio
|