Skip to content

Commit b9d4ecf

Browse files
authored
Replace token_list with generated_text in saved outputs (#12)
1 parent e04302c commit b9d4ecf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

benchmarks/benchmark_serving.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ class InputRequest:
9292
class RequestFuncOutput:
9393
input_request: InputRequest = None
9494
generated_token_list: list[str] = None
95+
generated_text: str = None
9596
success: bool = False
9697
latency: float = 0
9798
ttft: float = 0
@@ -102,7 +103,7 @@ def to_dict(self):
102103
return {
103104
"prompt": self.input_request.prompt,
104105
"original_output": self.input_request.output,
105-
"generated_token_list": self.generated_token_list,
106+
"generated_text": self.generated_text,
106107
"success": self.success,
107108
"latency": self.latency,
108109
"prompt_len": self.prompt_len
@@ -280,6 +281,7 @@ async def send_request(
280281
output.ttft = ttft
281282
output.latency = latency
282283
output.generated_token_list = generated_token_list
284+
output.generated_text = "".join(generated_token_list)
283285
output.success = True
284286
if pbar:
285287
pbar.update(1)

0 commit comments

Comments
 (0)