Spaces:

Vision-CAIR
/

MiniGPT-v2

Running on A10G

Vision-CAIR commited on Oct 19, 2023

Commit

bfcaa95

•

1 Parent(s): 15309b3

Update minigpt4/conversation/conversation.py

Files changed (1) hide show

minigpt4/conversation/conversation.py CHANGED Viewed

@@ -210,6 +210,7 @@ class Chat:
         image_emb, _ = self.model.encode_img(image)
         img_list.append(image_emb)
     def upload_img(self, image, conv, img_list):
         conv.append_message(conv.roles[0], "<Img><ImageHere></Img>")

         image_emb, _ = self.model.encode_img(image)
         img_list.append(image_emb)
+        return img_list
     def upload_img(self, image, conv, img_list):
         conv.append_message(conv.roles[0], "<Img><ImageHere></Img>")