flashinfer.md 515 Bytes
Newer Older
Lianmin Zheng's avatar
Lianmin Zheng committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
## Flashinfer Mode

[`flashinfer`](https://github.com/flashinfer-ai/flashinfer) is a kernel library for LLM serving; we use it here to support our attention computation.

### Install flashinfer

```bash
git submodule update --init --recursive
pip install 3rdparty/flashinfer/python
```

### Run Sever With Flashinfer Mode

Add through `--model_mode` argument from the command line.

Example:

```bash
python -m sglang.launch_server --model-path meta-llama/Llama-2-7b-chat-hf --port 30000 --model-mode flashinfer
```