sync: update vllm-awq-qwen36-dflash-ampere.sh from dankserver

- Model: Avesed/Qwen3.6-35B-A3B-INT8-W8A8 (was AWQ 4bit)
- Image: vllm/vllm-openai:nightly (was :latest)
- Spec method: mtp with 2 tokens (was dflash with 12)
- Max model len: 200700 (was 32768)
- Max num seqs: 3 (was 1)
- Max batched tokens: 3072 (was 4096)
- GPU mem util: 0.97 (was 0.96)
- CPU offload: 30GB (was 40GB)
- Add --enable-prefix-caching
- Add --kv-cache-dtype auto
This commit is contained in:
Frank committed 2026-08-10 03:23:09 -07:00
1 parent 11d66c3823
commit ced4c6f46e
1 file changed
+11 -8
+11 -8
View File
@@ -10,7 +10,7 @@
# #
set -euo pipefail set -euo pipefail
MODEL="cyankiwi/Qwen3.6-35B-A3B-AWQ-4bit" MODEL="Avesed/Qwen3.6-35B-A3B-INT8-W8A8"
PORT=34401 PORT=34401
exec docker run --rm --name vllm-awq-qwen36-dflash-ampere \ exec docker run --rm --name vllm-awq-qwen36-dflash-ampere \
@@ -22,20 +22,23 @@ exec docker run --rm --name vllm-awq-qwen36-dflash-ampere \
-e PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True \ -e PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True \
-e VLLM_USE_V2_MODEL_RUNNER=0 \ -e VLLM_USE_V2_MODEL_RUNNER=0 \
-v ~/.cache/huggingface:/root/.cache/huggingface \ -v ~/.cache/huggingface:/root/.cache/huggingface \
vllm/vllm-openai:latest "${MODEL}" \ vllm/vllm-openai:nightly "${MODEL}" \
--served-model-name "frank-default-small" \ --served-model-name "frank-default-small" \
--max-num-seqs 1 \ --max-num-seqs 3 \
--max-model-len 32768 \ --max-model-len 200700 \
--tensor-parallel-size 1 \ --tensor-parallel-size 1 \
--enable-chunked-prefill \ --enable-chunked-prefill \
--max-num-batched-tokens 4096 \ --enable-prefix-caching \
--max-num-batched-tokens 3072 \
--performance-mode throughput \ --performance-mode throughput \
--enable-auto-tool-choice --tool-call-parser qwen3_xml \ --enable-auto-tool-choice --tool-call-parser qwen3_xml \
--reasoning-parser qwen3 \ --reasoning-parser qwen3 \
--trust-remote-code \ --trust-remote-code \
--gpu-memory-utilization 0.96 \ --gpu-memory-utilization 0.97 \
--cpu-offload-gb 40 \ --kv-cache-dtype auto \
--cpu-offload-gb 30 \
--cpu-offload-params "experts" \ --cpu-offload-params "experts" \
--default-chat-template-kwargs '{"enable_thinking": true, "preserve_thinking": true}' \ --default-chat-template-kwargs '{"enable_thinking": true, "preserve_thinking": true}' \
-O3 \ -O3 \
--speculative-config '{"method": "dflash", "model": "z-lab/Qwen3.6-35B-A3B-DFlash", "num_speculative_tokens": 12}' --speculative-config '{"method": "mtp", "num_speculative_tokens": 2}'