Skip to content

Commit 5e15fb1

Browse files
authored
fix web-ui (#2758)
1 parent 2fcda5b commit 5e15fb1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

swift/ui/llm_infer/llm_infer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -401,6 +401,8 @@ def send_message(cls, running_task, template_type, prompt: str, image, video, au
401401
if infer_request.messages[-1]['role'] != 'assistant':
402402
infer_request.messages.append({'role': 'assistant', 'content': ''})
403403
for chunk in stream_resp:
404+
if chunk[0] is None:
405+
continue
404406
stream_resp_with_history += chunk[0].choices[0].delta.content if chat else chunk.choices[0].text
405407
infer_request.messages[-1]['content'] = stream_resp_with_history
406408
yield '', cls._replace_tag_with_media(infer_request), gr.update(value=None), gr.update(

0 commit comments

Comments
 (0)