Skip to content

Commit

Permalink
Update text_generation.py
Browse files Browse the repository at this point in the history
remove redundant `.tolist()` since it was already done above
  • Loading branch information
pavel-esir authored Dec 20, 2024
1 parent 04d9728 commit 98fa527
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/llm_bench/task/text_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 98fa527

Please sign in to comment.