From c703b51f8e994e450d48e92109eda9e70420246b Mon Sep 17 00:00:00 2001
From: Sinan <shuriken209master@googlemail.com>
Date: Tue, 12 Mar 2024 16:17:09 +0100
Subject: [PATCH] Enables image drop-in
---
cli_chat.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/cli_chat.py b/cli_chat.py
index a2b5249..bbea14a 100644
--- a/cli_chat.py
+++ b/cli_chat.py
@@ -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()