From 21fed3756f0e2638ef6f8e53689ec2c9c88616ea Mon Sep 17 00:00:00 2001 From: sbalandi Date: Mon, 20 Oct 2025 16:23:26 +0100 Subject: [PATCH] [llm_bench] Fix text use cases in case of complex_model_types --- tools/llm_bench/llm_bench_utils/get_use_case.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tools/llm_bench/llm_bench_utils/get_use_case.py b/tools/llm_bench/llm_bench_utils/get_use_case.py index 7964fcd3ec..292d0a59f8 100644 --- a/tools/llm_bench/llm_bench_utils/get_use_case.py +++ b/tools/llm_bench/llm_bench_utils/get_use_case.py @@ -50,13 +50,13 @@ def safe_json_load(file_path: Path) -> Optional[dict]: def resolve_complex_model_types(config): model_type = config.get("model_type").lower().replace('_', '-') if model_type == "gemma3": - return "visual_text_gen", model_type + return USE_CASES["visual_text_gen"][0], model_type if model_type == "gemma3-text": - return "text_gen", model_type + return USE_CASES["text_gen"][0], model_type if model_type in ["phi4mm", "phi4-multimodal"]: - return "visual_text_gen", model_type + return USE_CASES["visual_text_gen"][0], model_type if model_type == "llama4": - return "visual_text_gen", model_type + return USE_CASES["visual_text_gen"][0], model_type return None, None