From 12cde0df99a09c60190453f2a3e095784adc9f9a Mon Sep 17 00:00:00 2001 From: YAMY <74099316+YAMY1234@users.noreply.github.com> Date: Sun, 4 Jan 2026 19:39:23 -0800 Subject: [PATCH] [SPEC_V2] Fix Acclen drop when enabling DP Attention for Spec-Overlap (#16310) --- python/sglang/srt/speculative/eagle_worker_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/speculative/eagle_worker_v2.py b/python/sglang/srt/speculative/eagle_worker_v2.py index 7ba55f714..e77ed320b 100644 --- a/python/sglang/srt/speculative/eagle_worker_v2.py +++ b/python/sglang/srt/speculative/eagle_worker_v2.py @@ -491,7 +491,7 @@ class EagleDraftWorker(BaseDraftWorker): draft_input = EagleDraftInput( hidden_states=batch_result.logits_output.hidden_states, num_tokens_per_batch=self.speculative_num_steps + 1, - num_tokens_for_logprob_per_batch=1, + num_tokens_for_logprob_per_batch=self.speculative_num_steps + 1, ) select_index = ( torch.arange(len(batch.seq_lens), device=self.device)