Enables image drop-in

This commit is contained in:
Sinan 2024-03-12 16:17:09 +01:00 committed by GitHub
parent 8d4d9a6ccf
commit c703b51f8e
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: while cur_img_idx < num_images:
try: try:
image_file = input(f"({cur_img_idx + 1}/{num_images}) Input the image file path: ") 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: except KeyboardInterrupt:
print() print()