From 2e48584b62b3b5823ef234eeeadab12b5ee6098b Mon Sep 17 00:00:00 2001 From: Bruce-x-1997 Date: Fri, 31 Oct 2025 15:12:21 +0800 Subject: [PATCH] [bugfix] set is_prefill_only=false when mixed_chunk (#10889) --- python/sglang/srt/managers/schedule_batch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/sglang/srt/managers/schedule_batch.py b/python/sglang/srt/managers/schedule_batch.py index 6e3756e5c..abef463e3 100644 --- a/python/sglang/srt/managers/schedule_batch.py +++ b/python/sglang/srt/managers/schedule_batch.py @@ -1415,6 +1415,7 @@ class ScheduleBatch(ScheduleBatchDisaggregationDecodeMixin): self.extend_num_tokens += running_bs # TODO (lianmin): Revisit this. It should be seq_len - 1 self.extend_logprob_start_lens.extend([0] * running_bs) + self.is_prefill_only = False def new_page_count_next_decode(self, selected_indices: Optional[List[int]] = None): page_size = self.token_to_kv_pool_allocator.page_size