From 98fd2ab1170f64824f7f4d07b3ee270f53e6cdd4 Mon Sep 17 00:00:00 2001
From: coobiw <bowenqu@stu.pku.edu.cn>
Date: Tue, 11 Jun 2024 02:24:01 +0800
Subject: [PATCH] modify multi-img prompt in

---
 cli_demo.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/cli_demo.py b/cli_demo.py
index 62539e7..a931595 100644
--- a/cli_demo.py
+++ b/cli_demo.py
@@ -241,7 +241,11 @@ def main():
         try:
             if first:
                 if '<ImageHere>' not in query:
-                    query = f'<Img>{"<ImageHere>" * len(image_paths)}</Img> ' + query
+                    # query = f'<Img>{"<ImageHere>" * len(image_paths)}</Img> ' + query
+                    img_query = ""
+                    for _ in image_paths:
+                        img_query += '<Img><ImageHere></Img>'
+                    query = img_query + query
                 first = False
             if args.cpu_only:
                 model.bfloat16()