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
|