farrosalferro24 commited on
Commit
07d53ab
β€’
1 Parent(s): bb6d6f0

Update model/processing_gecko.py

Browse files
Files changed (1) hide show
  1. model/processing_gecko.py +1 -1
model/processing_gecko.py CHANGED
@@ -149,7 +149,7 @@ class GeckoProcessor(ProcessorMixin):
149
 
150
  assert len(processed_images) == blocks + 1
151
 
152
- return self.image_processor(images=processed_images, return_tensors='pt')['pixel_values'], coords
153
 
154
 
155
  def preprocess_interleaved_images_and_text(
 
149
 
150
  assert len(processed_images) == blocks + 1
151
 
152
+ return self.image_processor(images=processed_images, return_tensors='pt', padding=True)['pixel_values'], coords
153
 
154
 
155
  def preprocess_interleaved_images_and_text(