AMajor/grok-1
1
0
Fork 0
mirror of https://github.com/xai-org/grok-1.git synced 2025-12-16 15:11:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update requirements.txt

Browse Source 
from jax[cuda12_pip] to
jax[cuda12-pip]
This commit is contained in:
TianJian
2024-03-19 22:56:28 +08:00
committed by GitHub
parent d6d9447e2d
commit d0dcf7ca00
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -1,4 +1,4 @@
dm_haiku==0.0.12 dm_haiku==0.0.12
jax[cuda12_pip]==0.4.25 -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
numpy==1.26.4 numpy==1.26.4
sentencepiece==0.2.0 sentencepiece==0.2.0