Fix duration formatting

RWKV · Jun 13, 2023 · fa0e40b · fa0e40b
1 parent e77e22c
commit fa0e40b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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')