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

[Fix] DeepSeek-OCR-2 bench_serving: fix processor loading (#28483)

Browse Source
This commit is contained in:
Chetan Kumar Verma
2026-06-17 15:29:37 -07:00
committed by GitHub
parent e053890b6f
commit 732b81d5b9
1 changed files with 8 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/benchmark/utils.py
+8 -12
View File
@@ -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,