Model: Fix rounding

generated_tokens is always a whole number.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2023-11-30 01:55:46 -05:00
parent e703c716ee
commit 8a5ac5485b

View File

@@ -401,7 +401,7 @@ class ModelContainer:
elapsed_time = last_chunk_time - start_time
initial_response = f"Response: {round(generated_tokens, 2)} tokens generated in {round(elapsed_time, 2)} seconds"
initial_response = f"Response: {generated_tokens} tokens generated in {round(elapsed_time, 2)} seconds"
itemization = []
extra_parts = []