From 45cd88cf9d8a6f69341afcb3084f5c7edf087854 Mon Sep 17 00:00:00 2001 From: Deyao Zhu Date: Sun, 15 Oct 2023 19:57:39 +0300 Subject: [PATCH] update a small bug when the input doesn't contain image --- demo_v2.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/demo_v2.py b/demo_v2.py index 4f66d53..8f450c9 100644 --- a/demo_v2.py +++ b/demo_v2.py @@ -476,8 +476,9 @@ def gradio_answer(chatbot, chat_state, img_list, temperature): def gradio_stream_answer(chatbot, chat_state, img_list, temperature): print('chat state', chat_state.get_prompt()) - if not isinstance(img_list[0], torch.Tensor): - chat.encode_img(img_list) + if len(img_list) > 0: + if not isinstance(img_list[0], torch.Tensor): + chat.encode_img(img_list) streamer = chat.stream_answer(conv=chat_state, img_list=img_list, temperature=temperature,