From 98fa527b91e3e00623e1fa73be6873fa8634d3fa Mon Sep 17 00:00:00 2001 From: Pavel Esir Date: Fri, 20 Dec 2024 15:28:15 +0100 Subject: [PATCH] Update text_generation.py remove redundant `.tolist()` since it was already done above --- tools/llm_bench/task/text_generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/llm_bench/task/text_generation.py b/tools/llm_bench/task/text_generation.py index 485de94996..1f0985a445 100644 --- a/tools/llm_bench/task/text_generation.py +++ b/tools/llm_bench/task/text_generation.py @@ -324,7 +324,7 @@ def token_printer(): num, iter_data, tm_list.tolist(), - inference_durations.tolist(), + inference_durations, warm_up=(num == 0), max_rss_mem=max_rss_mem_consumption, max_shared_mem=max_shared_mem_consumption,