Skip to content

Commit

Permalink
fix local variable 'output_ids' referenced before assignment
Browse files Browse the repository at this point in the history
  • Loading branch information
irexyc committed Jan 8, 2024
1 parent 7be96a0 commit 444944a
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions lmdeploy/serve/turbomind/chatbot.py
Original file line number Diff line number Diff line change
Expand Up @@ -610,6 +610,7 @@ def stream_consumer(postprocess, res_queue, session, n_input_token,
tuple: status, text, generated token number
"""
status, res, n_token = None, '', 0
output_ids = np.zeros((1, 1, 0), dtype=np.uint32)
while True:
result = res_queue.get()
if result is None:
Expand Down

0 comments on commit 444944a

Please sign in to comment.