[Deepseek V3.2] Fix Deepseek MTP in V1 mode (#15429)

2025-12-18 17:56:07 -08:00
parent 4b4050e27e
commit e88e75a9df
1 changed files with 1 additions and 1 deletions
--- a/python/sglang/srt/layers/attention/nsa_backend.py
+++ b/python/sglang/srt/layers/attention/nsa_backend.py
@@ -435,7 +435,7 @@ class NativeSparseAttnBackend(
                # after verification. Lengths vary per request based on how many tokens
                # were accepted.
                page_table = torch.repeat_interleave(
-                    page_table, repeats=extend_seq_lens_cpu, dim=0
+                    page_table, repeats=forward_batch.extend_seq_lens, dim=0
                )

        elif forward_batch.forward_mode.is_extend():