vllm
**这是本文档旧的修订版!**
vllm
参数解读
vLLM 是用于 LLM 高效推理和服务的高性能框架。以下是其关键启动参数(vllm serve 或 python -m vllm.entrypoints.openai.api_server)的分类详解:
模型与推理基础配置
--model:指定模型路径或 HuggingFace 模型名称(如meta-llama/Llama-2-7b-chat-hf)。--tokenizer:指定分词器名称或路径,默认与--model一致。--trust-remote-code:允许执行来自 HuggingFace 的自定义代码(如加载部分新推出的开源模型时必填)。--download-dir:指定 HuggingFace 模型权重的本地缓存与下载目录。--revision:指定 HuggingFace 模型的具体分支、标签或 Commit ID。
GPU 资源与并行度
--tensor-parallel-size(-tp):张量并行(Tensor Parallelism)的 GPU 数量。跨卡切分单模型时使用(如 8 卡单机设为8)。--pipeline-parallel-size(-pp):流水线并行(Pipeline Parallelism)的 GPU 数量。通常优先增加-tp,在跨节点或卡数极多时配合使用。--gpu-memory-utilization:指定 vLLM 可调用的 GPU 显存比例,范围0.0到1.0(默认约为0.9)。预留显存给 KV Cache,若出现 OOM 可调小至0.85或0.8。--max-model-len:模型最大上下文序列长度(包含输入与输出)。若显存不足,降低此参数可大幅减少 KV Cache 显存占用。--cpu-offload-gb:从 GPU 卸载到 CPU 内存的权重规模(以 GB 为单位),适用于显存严重不足的单卡环境。
量化与数据类型
--dtype:模型计算与权重的精度数据类型。可选auto、float16、bfloat16、float32。Ampere 架构(如 A100/H100/3090/4090)建议设为bfloat16或auto。--quantization(-q):指定量化方法。可选awq、gptq、squeezellm、fp8、bitsandbytes等。若模型已提前量化,传入对应类型即可。--kv-cache-dtype:指定 KV Cache 的数据类型。支持auto、fp8(如fp8_e4m3),使用 FP8 储存 KV 缓存可减少显存占用并显著提升并发能力。
性能与调度调优
--max-num-seqs:单次批处理(Batch)中处理的最大并发序列数(默认256)。--max-num-batched-tokens:单次批处理中处理的最大 Token 数量(默认2048)。--enforce-eager:强制禁用 CUDA Graph 并使用 Eager 执行模式。虽稍微降低推理性能,但能省下 CUDA Graph 占用的显存,且便于 Debug。--swap-space:CPU 内存中分配用于 KV Cache 抢占交换的空间大小(以 GB 为单位,默认4)。
服务与 API 配置
--host:绑定的服务 IP 地址,例如0.0.0.0(对外开放)或127.0.0.1(仅本地访问)。--port:服务监听端口,默认8000。--api-key:设置访问 API 所需的密钥,提升服务安全性。--served-model-name:在 API 响应(如/v1/models接口)中显示的替代模型名称,方便客户端识别。
典型启动命令示例
- 单卡运行 7B/13B 模型:
bash vllm serve meta-llama/Llama-2-7b-chat-hf \ --host 0.0.0.0 \ --port 8000 \ --gpu-memory-utilization 0.9 \ --max-model-len 4096 - 多卡张量并行 + 量化运行大模型:
bash vllm serve Qwen/Qwen2.5-72B-Instruct-AWQ \ --tensor-parallel-size 4 \ --quantization awq \ --dtype bfloat16 \ --host 0.0.0.0 \ --port 8000
vllm.1787726805.txt.gz · 最后更改: 由 chenyang
