From d707652aec1d7fa62b30d608ffd19bf3c917e011 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9s=20Marafioti?= Date: Wed, 18 Sep 2024 14:29:10 +0200 Subject: [PATCH] Update src/transformers/models/idefics3/image_processing_idefics3.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> --- .../models/idefics3/image_processing_idefics3.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/transformers/models/idefics3/image_processing_idefics3.py b/src/transformers/models/idefics3/image_processing_idefics3.py index 80571c1b13444f..fa037b45296420 100644 --- 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 = [