From 652ece0710e62aaae82122fcf30c5acbe19c26ab Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 22 Jun 2023 15:34:13 +0200 Subject: [PATCH] Skip `test_conditional_generation_pt_pix2struct` in Past CI (torch < 1.11) (#24417) fix Co-authored-by: ydshieh --- .../pipelines/test_pipelines_image_to_text.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/tests/pipelines/test_pipelines_image_to_text.py b/tests/pipelines/test_pipelines_image_to_text.py index 2a73206f1b..7514f17919 100644 --- a/tests/pipelines/test_pipelines_image_to_text.py +++ b/tests/pipelines/test_pipelines_image_to_text.py @@ -18,11 +18,24 @@ import requests from transformers import MODEL_FOR_VISION_2_SEQ_MAPPING, TF_MODEL_FOR_VISION_2_SEQ_MAPPING, is_vision_available from transformers.pipelines import pipeline -from transformers.testing_utils import is_pipeline_test, require_tf, require_torch, require_vision, slow +from transformers.testing_utils import ( + is_pipeline_test, + is_torch_available, + require_tf, + require_torch, + require_vision, + slow, +) from .test_pipelines_common import ANY +if is_torch_available(): + from transformers.pytorch_utils import is_torch_greater_or_equal_than_1_11 +else: + is_torch_greater_or_equal_than_1_11 = False + + if is_vision_available(): from PIL import Image else: @@ -204,6 +217,9 @@ class ImageToTextPipelineTests(unittest.TestCase): with self.assertRaises(ValueError): outputs = pipe([image, image], prompt=[prompt, prompt]) + @unittest.skipIf( + not is_torch_greater_or_equal_than_1_11, reason="`Pix2StructImageProcessor` requires `torch>=1.11.0`." + ) @slow @require_torch def test_conditional_generation_pt_pix2struct(self):