minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Ethan (Yusheng) Su <yushengsu.thu@gmail.com>
This commit is contained in:
Kurt Shuster
2026-04-27 20:34:28 +00:00
committed by GitHub
co-authored by Ethan Su
parent 28ee08c172
commit f34c20af86
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/multimodal/processors/kimi_k25.py
+5 -1
View File
@@ -312,7 +312,11 @@ class KimiGPUProcessorWrapper:
# Convert to medias format for Kimi's HF processor # Convert to medias format for Kimi's HF processor
kwargs["medias"] = [{"type": "image", "image": img} for img in images] 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"): def _get_gpu_norm_tensors(self, device="cuda"):
if self._gpu_norm_tensors is None: if self._gpu_norm_tensors is None: