Update tools/llm_bench/llm_bench_utils/ov_utils.py

Co-authored-by: Ilya Lavrenov <[email protected]>
openvinotoolkit · Jan 10, 2025 · e9c7ba9 · e9c7ba9
1 parent 1d48dd1
commit e9c7ba9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/llm_bench/llm_bench_utils/ov_utils.py b/tools/llm_bench/llm_bench_utils/ov_utils.py
@@ -367,7 +367,7 @@ def __init__(self, main_model_name="unet") -> types.NoneType:
             self.iteration_time = []
             self.start_time = time.perf_counter()
             self.duration = -1
-            self.main_model_name =  main_model_name
+            self.main_model_name = main_model_name
 
         def __call__(self, step, num_steps, latents):
             self.iteration_time.append(time.perf_counter() - self.start_time)