From 9371062ef3e9ff1751a9bde6015da7e8f52c1980 Mon Sep 17 00:00:00 2001 From: Rohit Kumar Singh <9626333+SKRohit@users.noreply.github.com> Date: Wed, 17 Jun 2026 10:29:18 +0530 Subject: [PATCH] Fix deep seek ocr2 image processing (#27884) Co-authored-by: github-actions[bot] --- python/sglang/benchmark/datasets/image.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/python/sglang/benchmark/datasets/image.py b/python/sglang/benchmark/datasets/image.py index 284127620..3ee638dac 100644 --- a/python/sglang/benchmark/datasets/image.py +++ b/python/sglang/benchmark/datasets/image.py @@ -125,6 +125,13 @@ def create_mm_data_row( medias=medias, return_tensors="pt", )["input_ids"].numel() + elif type(processor).__name__ == "DeepseekVLV2Processor": + result = processor( + conversations=prompt_str, + images=images, + inference_mode=True, + ) + prompt_len = result.input_ids.numel() else: prompt_len = processor( text=[prompt_str],