5.5. Qwen¶
Qwen3-32B¶
模型下载¶
url: Qwen3-32B
branch:
mastercommit id:
bc4962f6
将上述url设定的路径下的内容全部下载到Qwen3-32B文件夹中。
注:需要安装以下依赖:
python3 -m pip install transformers==4.53.2
python3 -m pip install triton==3.1.0
环境变量¶
export VLLM_USE_V1=0
export TORCHGCU_INDUCTOR_ENABLE=0
export PYTORCH_EFML_BASED_GCU_CHECK=1
export TORCH_ECCL_AVOID_RECORD_STREAMS=1
export VLLM_WORKER_MULTIPROC_METHOD=spawn
export VLLM_ATTENTION_BACKEND=XFORMERS
批量离线推理¶
python3 -m vllm_utils.benchmark_test \
--model [path of Qwen3-32B] \
--tensor-parallel-size 4 \
--max-model-len=40960 \
--output-len=128 \
--demo=te \
--dtype=bfloat16 \
--device gcu \
--trust-remote-code \
--disable-async-output-proc
性能测试¶
# 启动服务端
python3 -m vllm.entrypoints.openai.api_server \
--model [path of Qwen3-32B] \
--tensor-parallel-size 4 \
--max-model-len 131072 \
--disable-log-requests \
--gpu-memory-utilization 0.9 \
--rope-scaling '{"rope_type":"yarn","factor":4.0,"original_max_position_embeddings":32768}' \
--block-size=64 \
--dtype=bfloat16 \
--device gcu \
--enable-chunked-prefill \
--disable-async-output-proc
# 启动客户端
python3 -m vllm_utils.benchmark_serving \
--backend vllm \
--dataset-name random \
--model [path of Qwen3-32B] \
--num-prompts 32 \
--random-input-len 1000 \
--random-output-len 700 \
--trust-remote-code \
--ignore_eos \
--strict-in-out-len \
--keep-special-tokens
注:
本模型支持的
max-model-len为131072;random-input-len、random-output-len和num-prompts可按需调整;
QWen3-30B-A3B¶
模型下载¶
url: QWen3-30B-A3B
branch:
mastercommit id:
e34b3e98
将上述url设定的路径下的内容全部下载到QWen3-30B-A3B文件夹中。
注:需要安装以下依赖:
python3 -m pip install transformers==4.53.2
python3 -m pip install triton==3.1.0
环境变量¶
export VLLM_USE_V1=0
export TORCHGCU_INDUCTOR_ENABLE=0
export PYTORCH_EFML_BASED_GCU_CHECK=1
export TORCH_ECCL_AVOID_RECORD_STREAMS=1
export VLLM_WORKER_MULTIPROC_METHOD=spawn
export VLLM_ATTENTION_BACKEND=XFORMERS
批量离线推理¶
python3 -m vllm_utils.benchmark_test \
--model [path of QWen3-30B-A3B] \
--tensor-parallel-size=2 \
--max-model-len=32768 \
--output-len=128 \
--demo=te \
--dtype=bfloat16 \
--device gcu \
--disable-async-output-proc
性能测试¶
# 启动服务端
python3 -m vllm.entrypoints.openai.api_server \
--model [path of QWen3-30B-A3B] \
--tensor-parallel-size 4 \
--max-model-len 131072 \
--disable-log-requests \
--block-size=64 \
--dtype=bfloat16 \
--device gcu \
--trust-remote-code \
--gpu-memory-utilization=0.9 \
--no-enable-prefix-caching \
--max-seq-len-to-capture 131072 \
--enable-chunked-prefill \
--rope-scaling '{"rope_type":"yarn","factor":4.0,"original_max_position_embeddings":32768}' \
--disable-async-output-proc
# 启动客户端
python3 -m vllm_utils.benchmark_serving \
--backend vllm \
--dataset-name random \
--model [path of QWen3-30B-A3B] \
--num-prompts 1 \
--random-input-len 130048 \
--random-output-len 1024 \
--trust-remote-code \
--ignore_eos \
--strict-in-out-len \
--keep-special-tokens
注:
本模型支持的
max-model-len为131072;random-input-len、random-output-len和num-prompts可按需调整;
QwQ-32B¶
模型下载¶
url: QwQ-32B
branch:
mastercommit id:
887ddbd72be5bed61cd702b87cd2fc25e65a708d
将上述url设定的路径下的内容全部下载到QwQ-32B文件夹中。
注:需要安装以下依赖:
python3 -m pip install transformers==4.53.2
python3 -m pip install triton==3.1.0
环境变量¶
export VLLM_USE_V1=0
export TORCHGCU_INDUCTOR_ENABLE=0
export PYTORCH_EFML_BASED_GCU_CHECK=1
export TORCH_ECCL_AVOID_RECORD_STREAMS=1
export VLLM_WORKER_MULTIPROC_METHOD=spawn
export VLLM_ATTENTION_BACKEND=XFORMERS
批量离线推理¶
python3.10 -m vllm_utils.benchmark_test \
--model=[path of QwQ-32B] \
--demo=te \
--tensor-parallel-size 4 \
--max-model-len=32768 \
--output-len=1024 \
--dtype=bfloat16 \
--device gcu \
--num-prompts 1 \
--block-size=64 \
--gpu-memory-utilization 0.9 \
--trust-remote-code \
--disable-async-output-proc
性能测试¶
# 启动服务端
python3 -m vllm.entrypoints.openai.api_server \
--model [path of QwQ-32B] \
--num-scheduler-steps=16 \
--tensor-parallel-size 4 \
--max-seq-len-to-capture=32768 \
--max-model-len 32768 \
--disable-log-requests \
--gpu-memory-utilization 0.9 \
--block-size=64 \
--dtype=bfloat16 \
--disable-async-output-proc
# 启动客户端
python3 -m vllm_utils.benchmark_serving \
--backend vllm \
--dataset-name random \
--model [path of QwQ-32B] \
--num-prompts 1 \
--random-input-len 512 \
--random-output-len 512 \
--trust-remote-code \
--ignore_eos \
--strict-in-out-len \
--keep-special-tokens
注:
本模型支持的
max-model-len为131072;input-len、output-len和num-prompts可按需调整;