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