Skip to content

Commit

Permalink
🐛 fixup streaming output length
Browse files Browse the repository at this point in the history
Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
  • Loading branch information
joerunde committed Apr 9, 2024
1 parent 8563f76 commit 0fc323c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions vllm/entrypoints/grpc/grpc_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,8 +236,8 @@ async def GenerateStream(

last_output_length = len(output.text)
last_token_count = len(output.token_ids)
# Accumulate full output for logging
full_output += output.text
# Save full output for logging
full_output = output.text

# Edit up the first_response for logging purposes only
if first_response is None:
Expand Down

0 comments on commit 0fc323c

Please sign in to comment.