[VLM] Fix Kimi-K2.5 CPU path: rename grid_thws -> image_grid_thw (#23501)

Co-authored-by: Ethan (Yusheng) Su <yushengsu.thu@gmail.com>
2026-04-27 20:34:28 +00:00
parent 28ee08c172
commit f34c20af86
1 changed files with 5 additions and 1 deletions
@@ -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: