Update processing_uform_gen.py
Browse files- processing_uform_gen.py +1 -1
processing_uform_gen.py
CHANGED
|
@@ -136,7 +136,7 @@ class VLMProcessor(ProcessorMixin):
|
|
| 136 |
image_features[i] = self.feature_extractor(image)
|
| 137 |
|
| 138 |
else:
|
| 139 |
-
image_features = self.
|
| 140 |
|
| 141 |
if text is not None and images is not None:
|
| 142 |
encoding["images"] = image_features
|
|
|
|
| 136 |
image_features[i] = self.feature_extractor(image)
|
| 137 |
|
| 138 |
else:
|
| 139 |
+
image_features = self.feature_extractor(images).unsqueeze(0)
|
| 140 |
|
| 141 |
if text is not None and images is not None:
|
| 142 |
encoding["images"] = image_features
|