andrewqian123
/

LLAMA_BATCH

Visual Question Answering

Model card Files Files and versions Community

andrewqian123 commited on Sep 3

Commit

35f8d11

•

1 Parent(s): a03d6de

Update modeling_minicpmv.py

Files changed (1) hide show

modeling_minicpmv.py +3 -2

modeling_minicpmv.py CHANGED Viewed

@@ -299,8 +299,9 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
             # copy_msgs[0]['content'] = '(<image>./</image>)\n' + copy_msgs[0]['content']
             #copy_msgs[0]["content"] = [image, copy_msgs[0]["content"]]
             #copy_msgs[0]["content"] = [image, copy_msgs[0]["content"]]
-            copy_msgs[0]["content"] = [copy_msgs[0]["content"]]
-            copy_msgs[0]["content"].insert(-1, im)
         images = []
         for i, msg in enumerate(copy_msgs):
             role = msg["role"]

             # copy_msgs[0]['content'] = '(<image>./</image>)\n' + copy_msgs[0]['content']
             #copy_msgs[0]["content"] = [image, copy_msgs[0]["content"]]
             #copy_msgs[0]["content"] = [image, copy_msgs[0]["content"]]
+            for im in image:
+                copy_msgs[0]["content"] = [copy_msgs[0]["content"]]
+                copy_msgs[0]["content"].insert(-1, im)
         images = []
         for i, msg in enumerate(copy_msgs):
             role = msg["role"]