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: