modify multi-img prompt in

Coobiw · Jun 10, 2024 · 98fd2ab · 98fd2ab
1 parent 79792fd
commit 98fd2ab
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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()