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)