From f34c20af86af8b0b15189f9f4d3384ee4471752a Mon Sep 17 00:00:00 2001 From: Kurt Shuster Date: Mon, 27 Apr 2026 16:34:28 -0400 Subject: [PATCH] [VLM] Fix Kimi-K2.5 CPU path: rename grid_thws -> image_grid_thw (#23501) Co-authored-by: Ethan (Yusheng) Su --- python/sglang/srt/multimodal/processors/kimi_k25.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/multimodal/processors/kimi_k25.py b/python/sglang/srt/multimodal/processors/kimi_k25.py index b117e2ab4..9838ca510 100644 --- a/python/sglang/srt/multimodal/processors/kimi_k25.py +++ b/python/sglang/srt/multimodal/processors/kimi_k25.py @@ -312,7 +312,11 @@ class KimiGPUProcessorWrapper: # Convert to medias format for Kimi's HF processor kwargs["medias"] = [{"type": "image", "image": img} for img in images] - return self._hf_processor(text=[input_text], **kwargs) + out = self._hf_processor(text=[input_text], **kwargs) + grid_thws = out.pop("grid_thws", None) + if grid_thws is not None: + out["image_grid_thw"] = grid_thws + return out def _get_gpu_norm_tensors(self, device="cuda"): if self._gpu_norm_tensors is None: