Update instruction

This commit is contained in:
Szymon Tworkowski 2024-03-18 09:09:40 -07:00 committed by GitHub
parent 1dfcf10e9d
commit 48094e5c68
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2,7 +2,8 @@
This repository contains JAX example code for loading and running the Grok-1 open-weights model. 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` - [see Downloading the weights](Downloading-the-weights) Make sure to download the checkpoint and place `ckpt-0` directory in `checkpoint` - see [Downloading the weights](Downloading-the-weights)
Then, run Then, run
```shell ```shell