Fix requirements.txt

Zblocker64 · Mar 17, 2024 · e50578b · e50578b
1 parent be76c95
commit e50578b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,4 @@
 dm_haiku==0.0.12
--f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-jax[cuda12_pip]==0.4.25
+jax[cuda12_pip]==0.4.25 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 numpy==1.26.4
 sentencepiece==0.2.0