diff --git a/evals/benchmark/stresscli/locust/tokenresponse.py b/evals/benchmark/stresscli/locust/tokenresponse.py index b566c0a1..77f9740b 100644 --- a/evals/benchmark/stresscli/locust/tokenresponse.py +++ b/evals/benchmark/stresscli/locust/tokenresponse.py @@ -26,7 +26,9 @@ def respStatics(environment, req, resp): num_token_input_prompt = -1 num_token_output = len( - tokenizer.encode(resp["response_string"].lstrip().encode("utf-8").decode("unicode_escape"), add_special_tokens=False) + tokenizer.encode( + resp["response_string"].lstrip().encode("utf-8").decode("unicode_escape"), add_special_tokens=False + ) ) return {