Update README.md

This commit is contained in:
Ross Nordeen 2024-03-18 02:55:11 +00:00 committed by system
parent 1f55fe3d8b
commit 8815da2d6b
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467

@ -1,21 +1,26 @@
--- ---
license: apache-2.0 license: apache-2.0
pipeline_tag: text-generation
library_name: grok-1
--- ---
# Grok-1 # Grok-1
This repository contains the weights of the Grok-1 open-weights model. This repository contains the weights of the Grok-1 open-weights model. You can find the code in the [GitHub Repository](https://github.com/xai-org/grok-1/tree/main).
Make sure to download the `int8` checkpoint to the `checkpoints` directory and run Make sure to download the `int8` checkpoint to the `checkpoints` directory.
```shell
huggingface-cli download xai-org/grok-1 --repo-type model --include ckpt/tensor* --local-dir checkpoints/ckpt-0 --local-dir-use-symlinks False
```
Then, you can run:
```shell ```shell
pip install -r requirements.txt pip install -r requirements.txt
python transformer.py
``` ```
to test the code.
You should be seeing output from the language model. 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. Due to the large size of the model (314B parameters), a multi-GPU machine is required to test the model with the example code.
p.s. we're hiring: https://x.ai/careers p.s. we're hiring: https://x.ai/career