File size: 1,080 Bytes
04deaba 1b15dde 04deaba 1b15dde 04deaba 1b15dde 04deaba 1b15dde 04deaba 1b15dde |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
## Usage: Serving with SGLang
- Download the weights. You can replace `/local/grok-2` with any other folder name you prefer.
```
hf download xai-org/grok-2 --local-dir /local/grok-2
```
You might encounter some errors during the download. Please retry until the download is successful.
If the download succeeds, the folder should contain **42 files** and be approximately 500 GB.
- Launch a server.
Install the latest SGLang inference engine by following the instructions at https://docs.sglang.ai/get_started/install.html.
Then, you can launch an inference server. This checkpoint is TP=8, so you will need 8 GPUs (each with > 40GB of memory).
```
python3 -m sglang.launch_server --model /local/grok-2 --tokenizer-path /local/grok-2/tokenizer.tok.json --tp 8 --quantization fp8 --attention-backend triton
```
- Send a request.
```
python3 -m sglang.test.send_one --prompt "Human: What is your name? <|separator|>\n\nAssistant:"
```
Learn more about other ways of sending requests [here](https://docs.sglang.ai/basic_usage/send_request.html). |