27 lines
1.6 KiB
Bash
27 lines
1.6 KiB
Bash
|
python -m vllm.entrypoints.openai.api_server \
|
|||
|
|
--model="/home/zwj/workspace/models/chatglm3-6b" \
|
|||
|
|
--tensor-parallel-size 1 \
|
|||
|
|
--trust-remote-code \
|
|||
|
|
--device auto \
|
|||
|
|
--gpu-memory-utilization 0.98 \
|
|||
|
|
--dtype half \
|
|||
|
|
--served-model-name "vllm" \
|
|||
|
|
--host 0.0.0.0 \
|
|||
|
|
--port 8080
|
|||
|
|
|
|||
|
|
# https://docs.vllm.ai/en/latest/serving/openai_compatible_server.html#named-arguments
|
|||
|
|
# model 模型路径，以文件夹结尾
|
|||
|
|
# tensor-parallel-size 张量并行副本数，即GPU的数量，咱这儿只有2张卡
|
|||
|
|
# trust-remote-code 信任远程代码，主要是为了防止模型初始化时不能执行仓库中的源码，默认值是False
|
|||
|
|
# device 用于执行 vLLM 的设备。可选auto、cuda、neuron、cpu
|
|||
|
|
# gpu-memory-utilization 用于模型推理过程的显存占用比例，范围为0到1。例如0.5表示显存利用率为 50%。如果未指定，则将使用默认值 0.9。
|
|||
|
|
# dtype “auto”将对 FP16 和 FP32 型使用 FP16 精度，对 BF16 型使用 BF16 精度。
|
|||
|
|
# “half”指FP16 的“一半”。推荐用于 AWQ 量化模型。
|
|||
|
|
# “float16”与“half”相同。
|
|||
|
|
# “bfloat16”用于在精度和范围之间取得平衡。
|
|||
|
|
# “float”是 FP32 精度的简写。
|
|||
|
|
# “float32”表示 FP32 精度。
|
|||
|
|
# kv-cache-dtype kv 缓存存储的数据类型。如果为“auto”，则将使用模型默认的数据类型。CUDA 11.8及以上版本 支持 fp8 （=fp8_e4m3） 和 fp8_e5m2。ROCm （AMD GPU） 支持 fp8 （=fp8_e4m3）
|
|||
|
|
# served-model-name 对外提供的API中的模型名称
|
|||
|
|
# host 监听的网络地址，0.0.0.0表示所有网卡的所有IP，127.0.0.1表示仅限本机
|
|||
|
|
# port API服务的端口
|