|
|
@@ -1,6 +1,6 @@
|
|
|
engine = "vllm"
|
|
|
name = "DeepSeek V4 Flash (vLLM)"
|
|
|
-docker_image = "aidendle94/sparkrun-vllm-ds4-gb10:production-ready"
|
|
|
+docker_image = "aidendle94/sparkrun-vllm-ds4-gb10:production-v2"
|
|
|
model = "deepseek-ai/DeepSeek-V4-Flash"
|
|
|
required_nodes = 2
|
|
|
tensor_parallel_size = 2
|
|
|
@@ -16,27 +16,32 @@ entrypoint = ["vllm", "serve"]
|
|
|
args = [
|
|
|
"--kv-cache-dtype", "fp8",
|
|
|
"--block-size", "256",
|
|
|
- "--max-model-len", "1048576",
|
|
|
- "--max-num-seqs", "6",
|
|
|
+ "--max-model-len", "524288",
|
|
|
+ "--max-num-seqs", "8",
|
|
|
"--max-num-batched-tokens", "8192",
|
|
|
- "--gpu-memory-utilization", "0.82",
|
|
|
+ "--enable-chunked-prefill",
|
|
|
+ "--gpu-memory-utilization", "0.85",
|
|
|
+ "--compilation-config", '{"cudagraph_mode":"FULL_AND_PIECEWISE","custom_ops":["all"]}',
|
|
|
"--enable-prefix-caching",
|
|
|
+ "--load-format", "safetensors",
|
|
|
"--speculative-config", '{"method":"mtp","num_speculative_tokens":2}',
|
|
|
"--tokenizer-mode", "deepseek_v4",
|
|
|
"--tool-call-parser", "deepseek_v4",
|
|
|
"--enable-auto-tool-choice",
|
|
|
"--reasoning-parser", "deepseek_v4",
|
|
|
- "--reasoning-config", '{"reasoning_parser":"deepseek_v4","reasoning_start_str":"","reasoning_end_str":""}',
|
|
|
- "--default-chat-template-kwargs.thinking", "true",
|
|
|
- "--default-chat-template-kwargs.reasoning_effort", "high",
|
|
|
+ "--default-chat-template-kwargs", '{"thinking":true,"preserve_thinking":true}',
|
|
|
"--enable-flashinfer-autotune",
|
|
|
"--trust-remote-code",
|
|
|
"--served-model-name", "deepseek-v4-flash", "default",
|
|
|
]
|
|
|
|
|
|
+# --distributed-executor-backend mp
|
|
|
+
|
|
|
[env]
|
|
|
-# HuggingFace / cache config
|
|
|
-VLLM_CACHE_ROOT = "/cache/huggingface/vllm-cache"
|
|
|
+# Ray memory management — prevents OOM during cudagraph compilation
|
|
|
+RAY_memory_monitor_refresh_ms = "0"
|
|
|
+RAY_num_prestart_python_workers = "0"
|
|
|
+RAY_object_store_memory = "1073741824"
|
|
|
|
|
|
# vLLM features for DeepSeek V4 on Blackwell (GB10 / sm_120)
|
|
|
VLLM_ALLOW_LONG_MAX_MODEL_LEN = "1"
|