Hugging Face's logo

Spaces:
mPLUG
/
TinyChart-3B
Running on Zero

App Files Community
xzl12306 commited on
Commit
328b8f6
·
verified ·
1 Parent(s): fae44e0

Update tinychart/mm_utils.py

Browse files
Files changed (1) hide show
  1. tinychart/mm_utils.py +1 -1
tinychart/mm_utils.py CHANGED
@@ -33,7 +33,7 @@ def process_images(images, image_processor, model_cfg):
33
  if image_aspect_ratio == 'pad':
34
  for image in images:
35
  image = expand2square(image, tuple(int(x*255) for x in image_processor.image_mean))
36
- image = image_processor.preprocess(image, return_tensors='pt')['pixel_values'][0]
37
  new_images.append(image)
38
  elif image_aspect_ratio == "anyres":
39
  for image in images:
 
33
  if image_aspect_ratio == 'pad':
34
  for image in images:
35
  image = expand2square(image, tuple(int(x*255) for x in image_processor.image_mean))
36
+ image = image_processor.preprocess(image, return_tensors='pt', padding=True)['pixel_values'][0]
37
  new_images.append(image)
38
  elif image_aspect_ratio == "anyres":
39
  for image in images: