chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 02:20:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

update a small bug when the input doesn't contain image

Browse Source
This commit is contained in:
Deyao Zhu 2023-10-15 19:57:39 +03:00
parent ffaf58e7f6
commit 45cd88cf9d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
demo_v2.py
View File

@ -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,