Fix requirements.txt

This commit is contained in:
Igor Babuschkin 2024-03-17 13:28:50 -07:00
parent be76c959fa
commit e50578b5f5

View File

@ -1,5 +1,4 @@
dm_haiku==0.0.12 dm_haiku==0.0.12
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html jax[cuda12_pip]==0.4.25 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jax[cuda12_pip]==0.4.25
numpy==1.26.4 numpy==1.26.4
sentencepiece==0.2.0 sentencepiece==0.2.0