From 732b81d5b9d36fee96fc922065af8a65d249b411 Mon Sep 17 00:00:00 2001
From: Chetan Kumar Verma <39086835+ckvermaAI@users.noreply.github.com>
Date: Thu, 18 Jun 2026 03:59:37 +0530
Subject: [PATCH] [Fix] DeepSeek-OCR-2 bench_serving: fix processor loading
 (#28483)

---
 python/sglang/benchmark/utils.py | 20 ++++++++------------
 1 file changed, 8 insertions(+), 12 deletions(-)

diff --git a/python/sglang/benchmark/utils.py b/python/sglang/benchmark/utils.py
index 031d38cdf..c3cbebf0c 100644
--- a/python/sglang/benchmark/utils.py
+++ b/python/sglang/benchmark/utils.py
@@ -71,21 +71,17 @@ def get_processor(
         pretrained_model_name_or_path is not None
         and pretrained_model_name_or_path != ""
     )
-    if pretrained_model_name_or_path.endswith(
-        ".json"
-    ) or pretrained_model_name_or_path.endswith(".model"):
-        from sglang.srt.utils.hf_transformers_utils import get_processor
 
-        return get_processor(pretrained_model_name_or_path)
-
-    if pretrained_model_name_or_path is not None and not os.path.exists(
-        pretrained_model_name_or_path
-    ):
-        pretrained_model_name_or_path = get_model(pretrained_model_name_or_path)
-    return AutoProcessor.from_pretrained(
-        pretrained_model_name_or_path, trust_remote_code=True
+    from sglang.srt.utils.hf_transformers_utils import (
+        get_processor as _srt_get_processor,
     )
 
+    if not pretrained_model_name_or_path.endswith(
+        (".json", ".model")
+    ) and not os.path.exists(pretrained_model_name_or_path):
+        pretrained_model_name_or_path = get_model(pretrained_model_name_or_path)
+    return _srt_get_processor(pretrained_model_name_or_path, trust_remote_code=True)
+
 
 def download_and_cache_hf_file(
     repo_id: str,