From 8815da2d6b54c6a32cb26df75a3336966fbae5ee Mon Sep 17 00:00:00 2001 From: Ross Nordeen Date: Mon, 18 Mar 2024 02:55:11 +0000 Subject: [PATCH] Update README.md --- README.md | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index f80ab9a..3861ac0 100644 --- a/README.md +++ b/README.md @@ -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