Fix missing return in processor.
Browse files- processor.py +1 -1
processor.py
CHANGED
|
@@ -56,7 +56,7 @@ class CondViTProcessor(ImageProcessingMixin):
|
|
| 56 |
"""
|
| 57 |
# Single Image
|
| 58 |
if isinstance(images, Image.Image):
|
| 59 |
-
|
| 60 |
data={"pixel_values": self.process_img(images), "texts": texts}
|
| 61 |
)
|
| 62 |
|
|
|
|
| 56 |
"""
|
| 57 |
# Single Image
|
| 58 |
if isinstance(images, Image.Image):
|
| 59 |
+
return BatchFeature(
|
| 60 |
data={"pixel_values": self.process_img(images), "texts": texts}
|
| 61 |
)
|
| 62 |
|