Optional texts in processor.

processor.py

@@ -38,7 +38,7 @@ class CondViTProcessor(ImageProcessingMixin):
         img = F.normalize(img, self.image_mean, self.image_std)
         return img
 
-    def __call__(self, images, texts):
+    def __call__(self, images, texts=None):
         """
         Parameters
         ----------
@@ -55,14 +55,15 @@ class CondViTProcessor(ImageProcessingMixin):
             texts : Union[str, List[str]]
         """
         # Single Image
+        data = {}
         if isinstance(images, Image.Image):
-            return BatchFeature(
-                data={"pixel_values": self.process_img(images), "texts": texts}
+            data["pixel_values"] = self.process_img(images)
+        else:
+            data["pixel_values"] = torch.stack(
+                [self.process_img(img) for img in images]
             )
 
-        return BatchFeature(
-            data={
-                "pixel_values": torch.stack([self.process_img(img) for img in images]),
-                "texts": texts,
-            }
-        )
+        if texts is not None:
+            data["texts"] = texts
+
+        return BatchFeature(data=data)