Update src/transformers/models/idefics3/image_processing_idefics3.py

Co-authored-by: amyeroberts <[email protected]>
huggingface · Sep 18, 2024 · 2967974 · 2967974
1 parent 215b636
commit 2967974
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/transformers/models/idefics3/image_processing_idefics3.py b/src/transformers/models/idefics3/image_processing_idefics3.py
@@ -885,10 +885,7 @@ def preprocess(
             ]
 
         if do_rescale:
-            rescaled_images_array = []
-            for image in images_list:
-                rescaled_images_array.append([rescale(img, rescale_factor, data_format=data_format) for img in image])
-            images_list = rescaled_images_array
+            images_list = [[self.rescale(img, rescale_factor, data_format=data_format) for image in images] for images in images_list]
 
         if do_normalize:
             images_list = [