Hyggge commited on
Commit
cfa4dbd
·
1 Parent(s): 0d9a24b

feat: update processing_valley.py

Browse files
Files changed (1) hide show
  1. processing_valley.py +1 -2
processing_valley.py CHANGED
@@ -251,7 +251,6 @@ class ValleyProcessor(ProcessorMixin):
251
  def __call__(self, messages, inference=True, **kwargs) -> BatchFeature:
252
  max_pixels=kwargs.get("max_pixels", self.max_pixels)
253
  min_pixels=kwargs.get("min_pixels", self.min_pixels)
254
- print(f"max_pixels: {max_pixels}, min_pixels: {min_pixels}")
255
  if max_pixels is not None:
256
  self.qwen2vl_image_processor.max_pixels = max_pixels
257
  if min_pixels is not None:
@@ -263,7 +262,7 @@ class ValleyProcessor(ProcessorMixin):
263
  elif type(messages["images"]) == str:
264
  images = [messages["images"]]
265
  else:
266
- images = messages["images"][:16] # support 16 images
267
 
268
  # Deal with conversations
269
  conversations = messages["conversations"]
 
251
  def __call__(self, messages, inference=True, **kwargs) -> BatchFeature:
252
  max_pixels=kwargs.get("max_pixels", self.max_pixels)
253
  min_pixels=kwargs.get("min_pixels", self.min_pixels)
 
254
  if max_pixels is not None:
255
  self.qwen2vl_image_processor.max_pixels = max_pixels
256
  if min_pixels is not None:
 
262
  elif type(messages["images"]) == str:
263
  images = [messages["images"]]
264
  else:
265
+ images = messages["images"]
266
 
267
  # Deal with conversations
268
  conversations = messages["conversations"]