xai-org/grok-1
xai-org/grok-1
1
1
Overview Code Discussions Pull Requests Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Szymon Tworkowski 2024-03-17 05:49:40 +00:00 committed by system
parent a1d5101bd6
commit 15af4e52e8
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
README.md
View File

@ -1,3 +1,19 @@
--- ---
license: apache-2.0 license: apache-2.0
--- ---
# Grok-1
This repository contains JAX example code for loading and running the Grok-1 open-weights model.
Make sure to download the `int8` checkpoint to the `checkpoints` directory and run
```shell
pip install -r requirements.txt
python transformer.py
```
to test the code.
You should be seeing output from the language model.
Due to the large size of the model (314B parameters), a multi-GPU machine is required to test the model with the example code.