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
pipeline_tag: text-generation
library_name: 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
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.
p.s. we're hiring: https://x.ai/careers
p.s. we're hiring: https://x.ai/career