diff --git a/model.py b/model.py index 872e4c1..3b2dc8d 100644 --- a/model.py +++ b/model.py @@ -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 = []