|
@@ -1,7 +1,7 @@
|
|
|
# DeepSeek-V4-Flash-DSpark — 2× GB10 / DGX Spark, TP=2 over RoCE.
|
|
# DeepSeek-V4-Flash-DSpark — 2× GB10 / DGX Spark, TP=2 over RoCE.
|
|
|
-# Uses the DSpark speculative decoding method with aidendle94's production-3.6 image.
|
|
|
|
|
|
|
+# Uses the DSpark speculative decoding method with aidendle94's production-3.7 image.
|
|
|
#
|
|
#
|
|
|
-# Image: aidendle94/sparkrun-vllm-ds4-gb10:production-3.6
|
|
|
|
|
|
|
+# Image: aidendle94/sparkrun-vllm-ds4-gb10:production-3.7
|
|
|
# Latest GB10-native image with DSpark support, b12x MoE, and multi-node fixes.
|
|
# Latest GB10-native image with DSpark support, b12x MoE, and multi-node fixes.
|
|
|
# Model: deepseek-ai/DeepSeek-V4-Flash-DSpark
|
|
# Model: deepseek-ai/DeepSeek-V4-Flash-DSpark
|
|
|
# The DSpark variant tuned for speculative decoding on DGX Spark.
|
|
# The DSpark variant tuned for speculative decoding on DGX Spark.
|
|
@@ -10,7 +10,7 @@
|
|
|
|
|
|
|
|
engine = "vllm"
|
|
engine = "vllm"
|
|
|
name = "DeepSeek V4 Flash DSpark (vLLM)"
|
|
name = "DeepSeek V4 Flash DSpark (vLLM)"
|
|
|
-docker_image = "aidendle94/sparkrun-vllm-ds4-gb10:production-3.6"
|
|
|
|
|
|
|
+docker_image = "aidendle94/sparkrun-vllm-ds4-gb10:production-3.7"
|
|
|
model = "deepseek-ai/DeepSeek-V4-Flash-DSpark"
|
|
model = "deepseek-ai/DeepSeek-V4-Flash-DSpark"
|
|
|
required_nodes = 2
|
|
required_nodes = 2
|
|
|
tensor_parallel_size = 2
|
|
tensor_parallel_size = 2
|
|
@@ -23,12 +23,12 @@ entrypoint = ["vllm", "serve"]
|
|
|
args = [
|
|
args = [
|
|
|
"--kv-cache-dtype", "fp8",
|
|
"--kv-cache-dtype", "fp8",
|
|
|
"--block-size", "256",
|
|
"--block-size", "256",
|
|
|
- "--max-model-len", "393216",
|
|
|
|
|
- "--max-num-seqs", "128",
|
|
|
|
|
- "--max-num-batched-tokens", "4096",
|
|
|
|
|
- "--max-cudagraph-capture-size", "128",
|
|
|
|
|
- "--gpu-memory-utilization", "0.8",
|
|
|
|
|
- "--served-model-name", "deepseek-v4-flash-dspark", "default",
|
|
|
|
|
|
|
+ "--max-model-len", "1048576",
|
|
|
|
|
+ "--max-num-seqs", "16",
|
|
|
|
|
+ "--max-num-batched-tokens", "16536",
|
|
|
|
|
+ "--max-cudagraph-capture-size", "16",
|
|
|
|
|
+ "--gpu-memory-utilization", "0.84",
|
|
|
|
|
+ "--served-model-name", "deepseek-v4-flash", "default",
|
|
|
# Engine backends
|
|
# Engine backends
|
|
|
"--attention-backend", "FLASHINFER_MLA_SPARSE_DSV4",
|
|
"--attention-backend", "FLASHINFER_MLA_SPARSE_DSV4",
|
|
|
"--moe-backend", "b12x",
|
|
"--moe-backend", "b12x",
|
|
@@ -43,25 +43,36 @@ args = [
|
|
|
"--async-scheduling",
|
|
"--async-scheduling",
|
|
|
"--no-scheduler-reserve-full-isl",
|
|
"--no-scheduler-reserve-full-isl",
|
|
|
# DSpark speculative decoding
|
|
# DSpark speculative decoding
|
|
|
|
|
+# "--speculative-config", """
|
|
|
|
|
+#'{"method":"dspark","num_speculative_tokens":5,"draft_sample_method":"probabilistic"}'""",
|
|
|
"--speculative-config", """
|
|
"--speculative-config", """
|
|
|
-'{"method":"dspark","num_speculative_tokens":3,"draft_sample_method":"probabilistic"}'""",
|
|
|
|
|
|
|
+'{"method":"dspark","num_speculative_tokens":7,"draft_sample_method":"greedy"}'""",
|
|
|
# Tokenizer / reasoning
|
|
# Tokenizer / reasoning
|
|
|
"--tokenizer-mode", "deepseek_v4",
|
|
"--tokenizer-mode", "deepseek_v4",
|
|
|
"--tool-call-parser", "deepseek_v4",
|
|
"--tool-call-parser", "deepseek_v4",
|
|
|
"--reasoning-parser", "deepseek_v4",
|
|
"--reasoning-parser", "deepseek_v4",
|
|
|
"--enable-auto-tool-choice",
|
|
"--enable-auto-tool-choice",
|
|
|
"--default-chat-template-kwargs.thinking=true",
|
|
"--default-chat-template-kwargs.thinking=true",
|
|
|
- "--default-chat-template-kwargs.reasoning_effort=max",
|
|
|
|
|
|
|
+ #"--default-chat-template-kwargs.reasoning_effort=max",
|
|
|
|
|
+ # temp 0.4 norm
|
|
|
|
|
+ "--override-generation-config", """
|
|
|
|
|
+'{"temperature": 0.4, "top_p": 1.0}'""",
|
|
|
|
|
+ "--default-chat-template-kwargs.reasoning_effort=high",
|
|
|
# Disable custom all-reduce (not needed on GB10 single-NIC)
|
|
# Disable custom all-reduce (not needed on GB10 single-NIC)
|
|
|
"--disable-custom-all-reduce",
|
|
"--disable-custom-all-reduce",
|
|
|
"--trust-remote-code",
|
|
"--trust-remote-code",
|
|
|
|
|
+ # From model card
|
|
|
|
|
+# does not work with b12x MoE
|
|
|
|
|
+# "--attention-config", """
|
|
|
|
|
+#'{"use_fp4_indexer_cache": true}'""",
|
|
|
]
|
|
]
|
|
|
|
|
|
|
|
[env]
|
|
[env]
|
|
|
# DSpark speculative decoding
|
|
# DSpark speculative decoding
|
|
|
VLLM_DSPARK_REPLICATE_MARKOV_W1 = "1"
|
|
VLLM_DSPARK_REPLICATE_MARKOV_W1 = "1"
|
|
|
# b12x MoE (this image's primary path)
|
|
# b12x MoE (this image's primary path)
|
|
|
-VLLM_USE_B12X_MOE = "1"
|
|
|
|
|
|
|
+# VLLM_USE_B12X_MOE = "1"
|
|
|
|
|
+VLLM_USE_B12X_MOE = "0" # cutlass decode
|
|
|
|
|
|
|
|
# b12x suite = REQUIRED by this image
|
|
# b12x suite = REQUIRED by this image
|
|
|
VLLM_USE_B12X_MHC = "1"
|
|
VLLM_USE_B12X_MHC = "1"
|