sgl-project/sglang

[Feature] support torchao for qwen2 models

Fermée

#2 219 ouverte le 27 nov. 2024

 (11 commentaires) (0 réaction) (0 personne assignée)Python (6 216 forks)auto 404
good first issuehelp wanted

Métriques du dépôt

Stars
 (28 442 étoiles)
Métriques de merge PR
 (Merge moyen 1j 13h) (1 000 PRs mergées en 30 j)

Description

I used one A30 card, and used Qwen2-7B-Instruct, the speed with quantization seems no different

python3 -m sglang.bench_latency --model ../Qwen2-7B-Instruct --batch-size 1 --input-len 200 --output-len 100 Benchmark ... Prefill. latency: 0.03508 s, throughput: 5700.84 token/s Decode. latency: 0.01952 s, throughput: 51.23 token/s Decode. latency: 0.01947 s, throughput: 51.37 token/s Decode. latency: 0.01939 s, throughput: 51.58 token/s Decode. latency: 0.01933 s, throughput: 51.74 token/s Decode. latency: 0.01928 s, throughput: 51.87 token/s Decode. median latency: 0.01924 s, median throughput: 51.98 token/s Total. latency: 1.942 s, throughput: 154.52 token/s

python3 -m sglang.bench_latency --model ../Qwen2-7B-Instruct --batch-size 1 --input-len 200 --output-len 100 --enable-torch-compile Benchmark ... Prefill. latency: 0.03655 s, throughput: 5471.84 token/s Decode. latency: 0.01852 s, throughput: 54.00 token/s Decode. latency: 0.01847 s, throughput: 54.14 token/s Decode. latency: 0.01845 s, throughput: 54.21 token/s Decode. latency: 0.01843 s, throughput: 54.26 token/s Decode. latency: 0.01838 s, throughput: 54.39 token/s Decode. median latency: 0.01836 s, median throughput: 54.46 token/s Total. latency: 1.855 s, throughput: 161.71 token/s

python3 -m sglang.bench_latency --model ../Qwen2-7B-Instruct --batch-size 1 --input-len 200 --output-len 100 --enable-torch-compile --torchao-config int8wo Benchmark ... Prefill. latency: 0.04469 s, throughput: 4475.31 token/s Decode. latency: 0.01860 s, throughput: 53.77 token/s Decode. latency: 0.01849 s, throughput: 54.09 token/s Decode. latency: 0.01844 s, throughput: 54.24 token/s Decode. latency: 0.01841 s, throughput: 54.32 token/s Decode. latency: 0.01837 s, throughput: 54.45 token/s Decode. median latency: 0.01836 s, median throughput: 54.46 token/s Total. latency: 1.863 s, throughput: 160.99 token/s

python3 -m sglang.bench_latency --model ../Qwen2-7B-Instruct --batch-size 1 --input-len 200 --output-len 100 --enable-torch-compile --torchao-config int4wo Benchmark ... Prefill. latency: 0.03558 s, throughput: 5621.52 token/s Decode. latency: 0.01855 s, throughput: 53.91 token/s Decode. latency: 0.01852 s, throughput: 54.01 token/s Decode. latency: 0.01845 s, throughput: 54.20 token/s Decode. latency: 0.01842 s, throughput: 54.28 token/s Decode. latency: 0.01841 s, throughput: 54.33 token/s Decode. median latency: 0.01837 s, median throughput: 54.44 token/s Total. latency: 1.855 s, throughput: 161.72 token/s

Guide contributeur