Add copied from

NielsRogge · Aug 31, 2024 · 2b77160 · 2b77160
1 parent 0adae27
commit 2b77160
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/src/transformers/models/auto/image_processing_auto.py b/src/transformers/models/auto/image_processing_auto.py
@@ -96,7 +96,7 @@
  ("layoutlmv2", ("LayoutLMv2ImageProcessor",)),
  ("layoutlmv3", ("LayoutLMv3ImageProcessor",)),
  ("levit", ("LevitImageProcessor",)),
- ("llava", ("CLIPImageProcessor",)),
+ ("llava", ("LlavaImageProcessor",)),
  ("llava_next", ("LlavaNextImageProcessor",)),
  ("llava_next_video", ("LlavaNextVideoImageProcessor",)),
  ("mask2former", ("Mask2FormerImageProcessor",)),

diff --git a/src/transformers/models/llava/image_processing_llava.py b/src/transformers/models/llava/image_processing_llava.py
@@ -214,6 +214,7 @@ def pad_to_square(
 
  return result
 
+ # Copied from transformers.models.clip.image_processing_clip.CLIPImageProcessor.resize
  def resize(
  self,
  image: np.ndarray,
@@ -389,10 +390,6 @@ def preprocess(
  # We assume that all images have the same channel dimension format.
  input_data_format = infer_channel_dimension_format(images[0])
 
- print("Input data format:", input_data_format)
- for image in images:
- print(image.shape)
-
  if do_pad:
  images = [
  self.pad_to_square(
@@ -403,10 +400,6 @@ def preprocess(
  for image in images
  ]
 
- print("After padding:")
- for image in images:
- print(image.shape)
-
  if do_resize:
  images = [
  self.resize(image=image, size=size, resample=resample, input_data_format=input_data_format)