AIDC-AI
/

Ovis-U1-3B

text-generation

image-text-to-text

Model card Files Files and versions

Flourish commited on Jul 3

Commit

26b2afa

·

verified ·

1 Parent(s): 9f8109f

Update modeling_ovis_u1.py

Files changed (1) hide show

modeling_ovis_u1.py +1 -1

modeling_ovis_u1.py CHANGED Viewed

@@ -397,7 +397,7 @@ class Aimv2VisualTokenizer(BaseVisualTokenizer):
         if multimodal_type == 'video':
             assert len(images) >= 1
         else:
-            assert len(images) == 1
         images = [image.convert("RGB") if convert_to_rgb and image.mode != 'RGB' else image for image in images ]
         # images = [np.array(image) for image in images]

         if multimodal_type == 'video':
             assert len(images) >= 1
         else:
+            pass
         images = [image.convert("RGB") if convert_to_rgb and image.mode != 'RGB' else image for image in images ]
         # images = [np.array(image) for image in images]