From dc64baaf3210a0378a8332f0ac31e9b4a6d77c11 Mon Sep 17 00:00:00 2001 From: wenkel1x Date: Sat, 11 Oct 2025 17:09:17 +0800 Subject: [PATCH] correct model name --- tools/llm_bench/llm_bench_utils/model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/llm_bench/llm_bench_utils/model_utils.py b/tools/llm_bench/llm_bench_utils/model_utils.py index 56800cbea8..c21ba161ec 100644 --- a/tools/llm_bench/llm_bench_utils/model_utils.py +++ b/tools/llm_bench/llm_bench_utils/model_utils.py @@ -298,7 +298,7 @@ def get_model_name(model_name_or_path, task=None): for use_case in possible_use_cases: for m_type in use_case.model_types: if model_name.lower().startswith(m_type): - return use_case, m_type + return use_case, model_name return None, None