From 069d4c577b39cc6fcd56f0ff138df7c8c793ea82 Mon Sep 17 00:00:00 2001 From: MARATRIX Date: Sat, 7 Mar 2026 08:14:02 +0800 Subject: [PATCH] Fix Kimi K2.5 PP layer range exposure for PD disaggregation (#19959) Signed-off-by: yafeng.li --- python/sglang/srt/models/kimi_k25.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/python/sglang/srt/models/kimi_k25.py b/python/sglang/srt/models/kimi_k25.py index 4b5dbbdfc..bf931d5cc 100644 --- a/python/sglang/srt/models/kimi_k25.py +++ b/python/sglang/srt/models/kimi_k25.py @@ -719,6 +719,14 @@ class KimiK25ForConditionalGeneration(nn.Module): pattern = MultiModalityDataPaddingPatternMultimodalTokens() return pattern.pad_input_tokens(input_ids, mm_inputs) + @property + def start_layer(self) -> int: + return self.language_model.start_layer + + @property + def end_layer(self) -> int: + return self.language_model.end_layer + def forward( self, input_ids: torch.Tensor,