From db15976fe9258adc1898e50e4671bfa3e3ae03f5 Mon Sep 17 00:00:00 2001 From: weireweire Date: Wed, 26 Aug 2026 14:18:58 +0800 Subject: [PATCH] Fix DSV4 DSpark sample-from-anchor initialization (#36419) Co-authored-by: weireweire <20922698+weireweire@users.noreply.github.com> --- python/sglang/srt/models/deepseek_v4_dspark.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/sglang/srt/models/deepseek_v4_dspark.py b/python/sglang/srt/models/deepseek_v4_dspark.py index d409dceb2..b826635b8 100644 --- a/python/sglang/srt/models/deepseek_v4_dspark.py +++ b/python/sglang/srt/models/deepseek_v4_dspark.py @@ -52,6 +52,7 @@ from sglang.srt.models.dspark import ( ) from sglang.srt.runtime_context import get_parallel from sglang.srt.speculative.dspark_components.dspark_config import ( + get_dspark_sample_from_anchor, parse_dspark_draft_config, ) from sglang.srt.speculative.ragged_verify import ( @@ -691,6 +692,7 @@ class DeepseekV4ForCausalLMDSpark(nn.Module): self.gamma = int( dspark_config.resolve_gamma(default=int(config.num_hidden_layers)) ) + self.sample_from_anchor = get_dspark_sample_from_anchor(config) self.block_size = self.gamma if dspark_config.target_layer_ids is not None: self.num_stages = len(dspark_config.target_layer_ids)