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,