From e50578b5f50e4c10c6e7cff31af1ef2bedb3beb8 Mon Sep 17 00:00:00 2001 From: Igor Babuschkin Date: Sun, 17 Mar 2024 13:28:50 -0700 Subject: [PATCH] Fix requirements.txt --- requirements.txt | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 02e1ce6..f6d124e 100644 --- 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