|
@@ -74,18 +74,26 @@ args = [
|
|
|
# Mirror of the lab's serve-mimo26-flash.sh: b12x linear + MoE backends.
|
|
# Mirror of the lab's serve-mimo26-flash.sh: b12x linear + MoE backends.
|
|
|
"--linear-backend", "b12x",
|
|
"--linear-backend", "b12x",
|
|
|
"--moe-backend", "b12x",
|
|
"--moe-backend", "b12x",
|
|
|
- "--load-format", "instanttensor",
|
|
|
|
|
|
|
+ # 2026-09-28 garble bisect, boot B1: mirror the lab launcher exactly —
|
|
|
|
|
+ # the b12x loader (plugin env below), full Omni + --mm-encoder-tp-mode
|
|
|
|
|
+ # weights instead of --language-model-only, explicit bf16 KV. instanttensor
|
|
|
|
|
+ # was our deviation; MiMo's split-GDN weight mapping is what the b12x
|
|
|
|
|
+ # loader path qualifies.
|
|
|
|
|
+ "--load-format", "b12x",
|
|
|
|
|
+ "--mm-encoder-tp-mode", "weights",
|
|
|
|
|
+ "--kv-cache-dtype", "bfloat16",
|
|
|
"--kernel-config", """
|
|
"--kernel-config", """
|
|
|
'{"enable_cutedsl_warmup":false,"enable_flashinfer_autotune":false}'""",
|
|
'{"enable_cutedsl_warmup":false,"enable_flashinfer_autotune":false}'""",
|
|
|
# DFlash draft from the checkpoint's dflash/ subfolder (7 tokens). The
|
|
# DFlash draft from the checkpoint's dflash/ subfolder (7 tokens). The
|
|
|
# karmic-kraken b12x supports non-causal attention, so the draft runs
|
|
# karmic-kraken b12x supports non-causal attention, so the draft runs
|
|
|
# B12X as well (the TRITON_ATTN workaround was only needed on r39).
|
|
# B12X as well (the TRITON_ATTN workaround was only needed on r39).
|
|
|
|
|
+ # DFlash-7 draft, B12X attention, kv_cache_dtype auto — the lab's
|
|
|
|
|
+ # Pro-TP8 DFlash launcher contract verbatim. The 2026-09-28 garble was
|
|
|
|
|
+ # NOT spec-related: instanttensor mis-mapped the split-GDN weights; the
|
|
|
|
|
+ # b12x loader fixed it (see --load-format comment).
|
|
|
"--speculative-config", """
|
|
"--speculative-config", """
|
|
|
-'{"method":"dflash","model":"{{dflash}}","num_speculative_tokens":7,"attention_backend":"B12X"}'""",
|
|
|
|
|
- # Nightly resolves the Omni wrapper's eagle3 hook on the inner language
|
|
|
|
|
- # model natively (no image patch needed); --language-model-only keeps
|
|
|
|
|
- # the visual/audio towers out of the memory budget.
|
|
|
|
|
- "--language-model-only",
|
|
|
|
|
|
|
+'{"method":"dflash","model":"{{dflash}}","num_speculative_tokens":7,"kv_cache_dtype":"auto","attention_backend":"B12X"}'""",
|
|
|
|
|
+ # (the --language-model-only line removed for boot B1 — see above)
|
|
|
"--reasoning-parser", "mimo",
|
|
"--reasoning-parser", "mimo",
|
|
|
"--tool-call-parser", "mimo",
|
|
"--tool-call-parser", "mimo",
|
|
|
"--enable-auto-tool-choice",
|
|
"--enable-auto-tool-choice",
|
|
@@ -119,6 +127,7 @@ container = "/root/.cache/vllm"
|
|
|
|
|
|
|
|
[env]
|
|
[env]
|
|
|
HF_HUB_OFFLINE = "1"
|
|
HF_HUB_OFFLINE = "1"
|
|
|
|
|
+VLLM_PLUGINS = "b12x_loader"
|
|
|
VLLM_ENABLE_ROCE_ALLREDUCE = "1"
|
|
VLLM_ENABLE_ROCE_ALLREDUCE = "1"
|
|
|
MAX_JOBS = "2"
|
|
MAX_JOBS = "2"
|
|
|
VLLM_ENGINE_READY_TIMEOUT_S = "3600"
|
|
VLLM_ENGINE_READY_TIMEOUT_S = "3600"
|