minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
b8zhong
2025-12-18 17:56:07 -08:00
committed by GitHub
parent 4b4050e27e
commit e88e75a9df
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/layers/attention/nsa_backend.py
+1 -1
View File
@@ -435,7 +435,7 @@ class NativeSparseAttnBackend(
# after verification. Lengths vary per request based on how many tokens # after verification. Lengths vary per request based on how many tokens
# were accepted. # were accepted.
page_table = torch.repeat_interleave( 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(): elif forward_batch.forward_mode.is_extend():