From 1d9274a5ffdaee36ced3426c03b71e2229326369 Mon Sep 17 00:00:00 2001 From: Nat Kershaw Date: Mon, 8 Apr 2024 16:36:16 -0700 Subject: [PATCH] Fix python lint #2 --- examples/python/model-qa.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/python/model-qa.py b/examples/python/model-qa.py index 8acec8596..c8362bc30 100644 --- a/examples/python/model-qa.py +++ b/examples/python/model-qa.py @@ -56,7 +56,7 @@ def main(args): if args.verbose: run_time = time.time() - started_timestamp - print(f"Prompt tokens: {len(input_tokens)}, New tokens: {len(new_tokens)}, Time to first: {(first_token_timestamp - started_timestamp):.2f}s, New tokens per second: {len(new_tokens)/run_time:.2f} tps") + print(f"Prompt length: {prompt_length}, New tokens: {len(new_tokens)}, Time to first: {(first_token_timestamp - started_timestamp):.2f}s, New tokens per second: {len(new_tokens)/run_time:.2f} tps") if __name__ == "__main__":