Skip to content

Commit

Permalink
LLM Bench: fix warning print
Browse files Browse the repository at this point in the history
  • Loading branch information
ilya-lavrenov committed Jan 13, 2025
1 parent b3096c3 commit 24a68b7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/llm_bench/llm_bench_utils/ov_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ def create_text_gen_model(model_path, device, **kwargs):
else:
if kwargs.get("genai", True) and is_genai_available(log_msg=True):
if model_class not in [OV_MODEL_CLASSES_MAPPING[default_model_type], OV_MODEL_CLASSES_MAPPING["mpt"], OV_MODEL_CLASSES_MAPPING["chatglm"]]:
log.warning("OpenVINO GenAI based benchmarking is not available for {model_type}. Will be switched to default benchmarking")
log.warning(f"OpenVINO GenAI based benchmarking is not available for {model_type}. Will be switched to default benchmarking")
else:
log.info("Selected OpenVINO GenAI for benchmarking")
return create_genai_text_gen_model(model_path, device, ov_config, **kwargs)
Expand Down

0 comments on commit 24a68b7

Please sign in to comment.