Update tools/llm_bench/task/text_generation.py

openvinotoolkit · Jan 16, 2025 · b6e607f · b6e607f
1 parent c862f07
commit b6e607f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/llm_bench/task/text_generation.py b/tools/llm_bench/task/text_generation.py
@@ -180,7 +180,7 @@ def run_text_generation(input_text, num, model, tokenizer, args, iter_data_list,
         if result_md5_list != prev_md5:
             log.warning(f"[{num}] Prompt[{prompt_index}]'s md5 {result_md5_list} "
                         f"is different from md5 of the {num - 1} iteration {prev_md5}")
-            metrics_print.print_generated(num, warm_up=(num == 0), generated=generated_text[0], prompt_idx=prompt_index))
+            metrics_print.print_generated(num, warm_up=(num == 0), generated=generated_text[0], prompt_idx=prompt_index)
     else:
         metrics_print.print_generated(num, warm_up=(num == 0), generated=generated_text[0], prompt_idx=prompt_index)
     if bench_hook is not None: