openbmb
/

MiniCPM-V-2_6

Image-Text-to-Text

feature-extraction

Model card Files Files and versions Community

HwwwH commited on Aug 7

Commit

5ff7973

•

1 Parent(s): 3a243ee

Update modeling_minicpmv.py

Files changed (1) hide show

modeling_minicpmv.py +2 -1

modeling_minicpmv.py CHANGED Viewed

@@ -292,7 +292,8 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
         if batched is False:
             images_list, msgs_list = [images_list], [msgs_list]
-        assert len(images_list) == len(msgs_list), "The batch dim of images_list and msgs_list should be the same."
         if processor is None:
             if self.processor is None:

         if batched is False:
             images_list, msgs_list = [images_list], [msgs_list]
+        if images_list is not None:
+            assert len(images_list) == len(msgs_list), "The batch dim of images_list and msgs_list should be the same."
         if processor is None:
             if self.processor is None: