diff --git a/python/sglang/srt/layers/attention/nsa_backend.py b/python/sglang/srt/layers/attention/nsa_backend.py index 18b1b9daf..4202501b1 100644 --- 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():