diff --git a/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_extend_npu_graph_runner.py b/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_extend_npu_graph_runner.py index fd230be5b..6d4077d4d 100644 --- a/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_extend_npu_graph_runner.py +++ b/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_extend_npu_graph_runner.py @@ -36,7 +36,7 @@ class EAGLEDraftExtendNpuGraphRunner(EAGLEDraftExtendCudaGraphRunner): def _replay_graph(self, shape_key, forward_batch): hf_config = self.model_runner.model_config.hf_config - if not (is_deepseek_dsa(hf_config) and is_deepseek_v4(hf_config)): + if not (is_deepseek_dsa(hf_config) or is_deepseek_v4(hf_config)): seq_lens = forward_batch.seq_lens_cpu.tolist() + [0] * ( self.bs - self.raw_bs ) diff --git a/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_npu_graph_runner.py b/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_npu_graph_runner.py index 006f46ee6..113eed0f5 100644 --- a/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_npu_graph_runner.py +++ b/python/sglang/srt/hardware_backend/npu/graph_runner/eagle_draft_npu_graph_runner.py @@ -88,7 +88,7 @@ class EAGLEDraftNpuGraphRunner(EAGLEDraftCudaGraphRunner): def _replay_graph(self, shape_key, forward_batch): hf_config = self.model_runner.model_config.hf_config - if not (is_deepseek_dsa(hf_config) and is_deepseek_v4(hf_config)): + if not (is_deepseek_dsa(hf_config) or is_deepseek_v4(hf_config)): seq_lens_for_each_draft_step = [] for speculative_step_id in range(self.speculative_num_steps - 1): seq_lens_cpu = (