Skip to content

Commit a7df37c

Browse files
authored
[llm_bench] Fix text use_case in case of complex_model_types (openvinotoolkit#2874)
## Description use_cases have become object, but for complex model types are remains text, fix it <!--- Jira ticket number (e.g., 123). Delete if there's no ticket. Don't include full link or project name. --> Ticket: CVS-174931 <!-- Remove if not applicable --> Fixes #(issue) ## Checklist: - [ ] Tests have been updated or added to cover the new code <!--- If the change isn't maintenance related, update the tests at https://github.com/openvinotoolkit/openvino.genai/tree/master/tests or explain in the description why the tests don't need an update. --> - [ ] This patch fully addresses the ticket. <!--- If follow-up pull requests are needed, specify in description. --> - [ ] I have made corresponding changes to the documentation
1 parent c894846 commit a7df37c

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tools/llm_bench/llm_bench_utils/get_use_case.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,13 +50,13 @@ def safe_json_load(file_path: Path) -> Optional[dict]:
5050
def resolve_complex_model_types(config):
5151
model_type = config.get("model_type").lower().replace('_', '-')
5252
if model_type == "gemma3":
53-
return "visual_text_gen", model_type
53+
return USE_CASES["visual_text_gen"][0], model_type
5454
if model_type == "gemma3-text":
55-
return "text_gen", model_type
55+
return USE_CASES["text_gen"][0], model_type
5656
if model_type in ["phi4mm", "phi4-multimodal"]:
57-
return "visual_text_gen", model_type
57+
return USE_CASES["visual_text_gen"][0], model_type
5858
if model_type == "llama4":
59-
return "visual_text_gen", model_type
59+
return USE_CASES["visual_text_gen"][0], model_type
6060
return None, None
6161

6262

0 commit comments

Comments
 (0)