Spaces:
Running
Running
Yaron Koresh
commited on
Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
@@ -11,9 +11,9 @@ transformers
|
|
11 |
pathos
|
12 |
dill
|
13 |
flax
|
14 |
-
--
|
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
|
|
|
11 |
pathos
|
12 |
dill
|
13 |
flax
|
14 |
+
--extra-index-url https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
|
|
15 |
jax[cuda_local]
|
16 |
+
jaxlib[cuda_local]
|
17 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
18 |
torch
|
19 |
torchaudio
|