Enables image drop-in (#12)

deepseek-ai · Mar 13, 2024 · 86a3096 · 86a3096
1 parent 8d4d9a6
commit 86a3096
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()