fix: allow pass image kwargs to image processor

#90
Files changed (1) hide show
  1. processing_paddleocr_vl.py +1 -1
processing_paddleocr_vl.py CHANGED
@@ -158,7 +158,7 @@ class PaddleOCRVLProcessor(ProcessorMixin):
158
  )
159
 
160
  if images is not None:
161
- image_inputs = self.image_processor(images=images, return_tensors="pt")
162
  image_inputs["pixel_values"] = image_inputs["pixel_values"]
163
  image_grid_thw = image_inputs["image_grid_thw"]
164
 
 
158
  )
159
 
160
  if images is not None:
161
+ image_inputs = self.image_processor(images=images, **output_kwargs["images_kwargs"])
162
  image_inputs["pixel_values"] = image_inputs["pixel_values"]
163
  image_grid_thw = image_inputs["image_grid_thw"]
164