Enables image drop-in (#12)

This commit is contained in:
Sinan 2024-03-13 09:16:32 +01:00 committed by GitHub
parent 8d4d9a6ccf
commit 86a309683b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -116,6 +116,7 @@ def chat(args, tokenizer, vl_chat_processor, vl_gpt, generation_config):
while cur_img_idx < num_images:
try:
image_file = input(f"({cur_img_idx + 1}/{num_images}) Input the image file path: ")
image_file = image_file.strip() # trim whitespaces around path, enables drop-in from for example Dolphin
except KeyboardInterrupt:
print()