VoVoR commited on
Commit
2bb78ca
1 Parent(s): aab59dc

Update processing_uform_gen.py

Browse files
Files changed (1) hide show
  1. 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.image_processor(images).unsqueeze(0)
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