AMajor/grok-1
1
0
Fork 0
mirror of https://github.com/xai-org/grok-1.git synced 2024-11-24 12:39:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

added jaxlib specific version

Browse Source 
In some cases, jaxlib will not install exact compatible version number.  specifying version in requirements avoids startup error
This commit is contained in:
Michael Lucas 2024-04-06 18:06:14 -05:00 committed by GitHub
parent 7050ed204b
commit 3577bf2688
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements.txt
View File

@ -1,4 +1,5 @@
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
jaxlib==0.4.25
numpy==1.26.4 numpy==1.26.4
sentencepiece==0.2.0 sentencepiece==0.2.0