test

visualcloze.py

@@ -339,6 +339,8 @@ class VisualClozeModel:
                     processed_images.append(blank)
                     if i == grid_h - 1:
                         mask_position.append(1)
+                    else:
+                        raise ValueError('Please provide each image in the in-context example.')
             
         return processed_images