nit change version deprecation warning

huggingface · Oct 17, 2024 · 7a4a0e4 · 7a4a0e4
1 parent 819e22c
commit 7a4a0e4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/src/transformers/pipelines/image_text_to_text.py b/src/transformers/pipelines/image_text_to_text.py
@@ -332,11 +332,10 @@ def __call__(
                     )
                 else:
                     # Reorganize the images to match the prompts
-                    images_reorganized = []
-                    for num_images in num_images_in_text:
-                        images_reorganized.append(images[:num_images])
-                        images = images[num_images:]
-                    images = images_reorganized
+                    images = [
+                        images[sum(num_images_in_text[:i]) : sum(num_images_in_text[: i + 1])]
+                        for i in range(len(num_images_in_text))
+                    ]
         else:
             if hasattr(self.processor, "image_token") and self.processor.image_token is not None:
                 logger.warning(

diff --git a/src/transformers/pipelines/image_to_text.py b/src/transformers/pipelines/image_to_text.py
@@ -135,7 +135,7 @@ def preprocess(self, image, prompt=None, timeout=None):
 
         if prompt is not None:
             logger.warning_once(
-                "Passing `prompt` to the `image-to-text` pipeline is deprecated and will be removed in version 4.45"
+                "Passing `prompt` to the `image-to-text` pipeline is deprecated and will be removed in version 4.48"
                 " of 🤗 Transformers. Use the `image-text-to-text` pipeline instead",
             )
             if not isinstance(prompt, str):