|
|
@@ -20,6 +20,9 @@
|
|
|
#
|
|
|
# Reference: Nvidia Developer Forum — DeepSeek-V4-Flash-0731 recipe
|
|
|
# https://forums.developer.nvidia.com/t/deepseek-v4-flash-0731-dspark-1m-nvfp4-kv-2x-dgx-spark/378824
|
|
|
+# Encoder reasoning-effort fix source: 0rand's serving stack (thread 372268/708)
|
|
|
+# https://forums.developer.nvidia.com/t/deepseek-v4-flash-aiden-recipe-from-reddit-1m-token-session-operational-cuda-12-1-tailored-for-dgx-spark-gb10/372268/708
|
|
|
+# (three-level low/high/max ladder; bundled into the :fixed overlay here)
|
|
|
|
|
|
engine = "vllm"
|
|
|
name = "DeepSeek V4 Flash DSpark (vLLM)"
|
|
|
@@ -38,10 +41,10 @@ args = [
|
|
|
"--kv-cache-dtype", "fp8",
|
|
|
"--block-size", "256",
|
|
|
"--load-format", "auto",
|
|
|
- "--max-model-len", "524288",
|
|
|
+ "--max-model-len", "1048576",
|
|
|
"--max-num-seqs", "16",
|
|
|
- "--max-num-batched-tokens", "16536",
|
|
|
- "--max-cudagraph-capture-size", "16",
|
|
|
+ "--max-num-batched-tokens", "16384",
|
|
|
+ "--max-cudagraph-capture-size", "512",
|
|
|
"--gpu-memory-utilization", "0.84",
|
|
|
"--served-model-name", "deepseek-v4-flash",
|
|
|
# Engine backends
|
|
|
@@ -68,11 +71,15 @@ args = [
|
|
|
"--tool-call-parser", "deepseek_v4",
|
|
|
"--reasoning-parser", "deepseek_v4",
|
|
|
"--enable-auto-tool-choice",
|
|
|
+ # Proven generation defaults (thread 372268/708 serving stack; with
|
|
|
+ # --generation-config vllm only the override params are applied, so these
|
|
|
+ # set the served defaults). repetition_penalty 1.0 = identity: the field is
|
|
|
+ # accepted by the spec but never deviates from no-op during req handling.
|
|
|
+ "--override-generation-config", """
|
|
|
+'{"temperature":0.8,"top_p":0.25,"repetition_penalty":1.0}'""",
|
|
|
"--generation-config", "vllm",
|
|
|
- "--default-chat-template-kwargs.thinking=true",
|
|
|
- "--default-chat-template-kwargs.reasoning_effort=high",
|
|
|
-# "--default-chat-template-kwargs.temperature=1",
|
|
|
-# "--default-chat-template-kwargs.top_p=1",
|
|
|
+ "--default-chat-template-kwargs.enable_thinking=true",
|
|
|
+ "--default-chat-template-kwargs.reasoning_effort=max",
|
|
|
# Disable custom all-reduce (not needed on GB10 single-NIC)
|
|
|
"--disable-custom-all-reduce",
|
|
|
"--trust-remote-code",
|
|
|
@@ -124,8 +131,13 @@ HF_HUB_OFFLINE = "1"
|
|
|
TORCH_CUDA_ARCH_LIST = "12.1a"
|
|
|
|
|
|
# NCCL overrides (cluster config ib_hca/socket_ifname used for most; recipe wins last)
|
|
|
+# NCCL_CROSS_NIC / NCCL_PROTO come from the cluster [nccl] section (2nodes-dac:
|
|
|
+# cross_nic=0, proto=LL,LL128,Simple) — matches the reference compose except
|
|
|
+# cross_nic (reference runs 1; our cluster config deliberately uses 0 for this
|
|
|
+# single-RoCE topology). NCCL_NVLS_ENABLE mirrors the reference compose.
|
|
|
NCCL_P2P_LEVEL = "SYS"
|
|
|
NCCL_CUMEM_ENABLE = "0"
|
|
|
+NCCL_NVLS_ENABLE = "0"
|
|
|
NCCL_IB_TIMEOUT = "24"
|
|
|
NCCL_IB_QPS_PER_CONNECTION = "8"
|
|
|
NCCL_BUFFERSIZE = "16777216"
|