From fa0e40b32d308c6a0c9e6e8b26022726110d927a Mon Sep 17 00:00:00 2001 From: saharNooby Date: Tue, 13 Jun 2023 18:16:20 +0400 Subject: [PATCH] Fix duration formatting --- rwkv/chat_with_bot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rwkv/chat_with_bot.py b/rwkv/chat_with_bot.py index 1241c3e..78cbecd 100644 --- a/rwkv/chat_with_bot.py +++ b/rwkv/chat_with_bot.py @@ -114,9 +114,9 @@ def split_last_end_of_line(tokens): process_tokens(split_last_end_of_line(prompt_tokens)) -processing_duration = (time.time() - processing_start) * 1000 +processing_duration = time.time() - processing_start -print(f'Processed in {processing_duration} ms, {processing_duration / prompt_token_count} ms per token') +print(f'Processed in {int(processing_duration)} s, {int(processing_duration / prompt_token_count * 1000)} ms per token') save_thread_state('chat_init') save_thread_state('chat')