Update README.md

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
-pip install -r requirements.txt
-python transformer.py
+huggingface-cli download xai-org/grok-1 --repo-type model --include ckpt/tensor* --local-dir checkpoints/ckpt-0 --local-dir-use-symlinks False
 ```
 
-to test the code.
+Then, you can run:
+
+```shell
+pip install -r requirements.txt
+```
 
 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