Shikong/grok-1
1
0
Fork 0
mirror of https://github.com/xai-org/grok-1 synced 2024-11-09 10:01:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix requirements.txt

Browse Source
This commit is contained in:
Igor Babuschkin 2024-03-17 13:28:50 -07:00
parent be76c959fa
commit e50578b5f5
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
requirements.txt
View File

@ -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