README.md 1.82 KB
Newer Older
Lianmin Zheng's avatar
Lianmin Zheng committed
1
2
3
4
5
6

# Benchmark Latency and Throughput

## SGLang

### Launch server
Lianmin Zheng's avatar
Lianmin Zheng committed
7
```
Lianmin Zheng's avatar
Lianmin Zheng committed
8
python -m sglang.launch_server --model-path meta-llama/Llama-2-7b-chat-hf --port 30000
Lianmin Zheng's avatar
Lianmin Zheng committed
9
```
Ying Sheng's avatar
Ying Sheng committed
10

Lianmin Zheng's avatar
Lianmin Zheng committed
11
### Benchmark one batch
Lianmin Zheng's avatar
Lianmin Zheng committed
12
13

```
Lianmin Zheng's avatar
Lianmin Zheng committed
14
15
python3 bench_one.py
python3 bench_one.py --batch-size 64
Lianmin Zheng's avatar
Lianmin Zheng committed
16
17
```

Lianmin Zheng's avatar
Lianmin Zheng committed
18
19
### Benchmark online serving with many requests

Lianmin Zheng's avatar
Lianmin Zheng committed
20
```
Lianmin Zheng's avatar
Lianmin Zheng committed
21
python3 bench_serving.py --backend srt --port 30000 --tokenizer meta-llama/Llama-2-7b-chat-hf --num-prompt 1000 --request-rate 100 --input-len 1024 --output-len 256
Lianmin Zheng's avatar
Lianmin Zheng committed
22
23
```

Lianmin Zheng's avatar
Lianmin Zheng committed
24
25
26
### Benchmark online serving on the ShareGPT dataset

#### Download data
Ying Sheng's avatar
Ying Sheng committed
27
```
Lianmin Zheng's avatar
Lianmin Zheng committed
28
wget https://huggingface.co/datasets/anon8231489123/ShareGPT_Vicuna_unfiltered/resolve/main/ShareGPT_V3_unfiltered_cleaned_split.json
Ying Sheng's avatar
Ying Sheng committed
29
30
```

Lianmin Zheng's avatar
Lianmin Zheng committed
31
32
33
34
35
36
#### Run ShareGPT
```
python3 bench_throughput.py --backend srt --port 30000 --tokenizer meta-llama/Llama-2-7b-chat-hf --dataset ShareGPT_V3_unfiltered_cleaned_split.json --num-prompts 10 --request-rate 10
```

## Other baselines
Lianmin Zheng's avatar
Lianmin Zheng committed
37
38
39

### vLLM
```
Ying Sheng's avatar
Ying Sheng committed
40
python3 -m vllm.entrypoints.api_server --model meta-llama/Llama-2-7b-chat-hf --tensor-parallel 1 --disable-log-requests --swap-space 16 --port 21000
Lianmin Zheng's avatar
Lianmin Zheng committed
41
42
43
```

```
Lianmin Zheng's avatar
Lianmin Zheng committed
44
45
# run synthetic
python3 bench_throughput.py --backend vllm --port 30000 --tokenizer meta-llama/Llama-2-7b-chat-hf --num-prompt 1000 --request-rate 100 --input-len 1024 --output-len 256
Lianmin Zheng's avatar
Lianmin Zheng committed
46
47
```

Ying Sheng's avatar
Ying Sheng committed
48
```
Lianmin Zheng's avatar
Lianmin Zheng committed
49
50
# run ShareGPT
python3 bench_throughput.py --backend vllm --port 21000 --tokenizer meta-llama/Llama-2-7b-chat-hf --dataset ShareGPT_V3_unfiltered_cleaned_split.json --num-prompts 10 --request-rate 10
Ying Sheng's avatar
Ying Sheng committed
51
52
```

Lianmin Zheng's avatar
Lianmin Zheng committed
53
54
55
56
57
58
59

### LightLLM
```
python -m lightllm.server.api_server --model_dir ~/model_weights/Llama-2-7b-chat-hf --max_total_token_num 15600 --tokenizer_mode auto --port 22000
```

```
Lianmin Zheng's avatar
Lianmin Zheng committed
60
61
python3 bench_throughput.py --backend lightllm --port 22000 --tokenizer meta-llama/Llama-2-7b-chat-hf --dataset ShareGPT_V3_unfiltered_cleaned_split.json --num-prompts 10 --request-rate 10
```