AMajor/grok-1
1
0
Fork 0
mirror of https://github.com/xai-org/grok-1.git synced 2025-12-14 22:21:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add initial code

Browse Source
This commit is contained in:
Igor Babuschkin
2024-03-14 15:03:58 -07:00
parent 5aabc78af1
commit be76c959fa
11 changed files with 2552 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
README.md
View File

@ -1,2 +1,31 @@
# grok-open
Open release of the Grok model
# Grok-1
This repository contains JAX example code for loading and running the Grok-1 open-weights model.
Make sure to download the checkpoint and place `ckpt-0` directory in `checkpoint`.
Then, run
```shell
pip install -r requirements.txt
python run.py
```
to test the code.
The script loads the checkpoint and samples from the model on a test input.
Due to the large size of the model (314B parameters), a machine with enough GPU memory is required to test the model with the example code.
The implementation of the MoE layer in this repository is not efficient. The implementation was chosen to avoid the need for custom kernels to validate the correctness of the model.
# Downloading the weights
You can download the weights using a torrent client and this magnet link:
```
magnet:?xt=urn:btih:5f96d43576e3d386c9ba65b883210a393b68210e&tr=https%3A%2F%2Facademictorrents.com%2Fannounce.php&tr=udp%3A%2F%2Ftracker.coppersurfer.tk%3A6969&tr=udp%3A%2F%2Ftracker.opentrackr.org%3A1337%2Fannounce
```
# License
The code and associated Grok-1 weights in this release are licensed under the
Apache 2.0 license. The license only applies to the source files in this
repository and the model weights of Grok-1.