From 059c8fd248ac23288f10c479df235f03b21358c0 Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Sun, 3 Jul 2022 10:36:30 +0800 Subject: [PATCH] [to #42322933] do not run imagen on gated for now --- tests/pipelines/test_text_to_image_synthesis.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/test_text_to_image_synthesis.py b/tests/pipelines/test_text_to_image_synthesis.py index d5ce990d..568b4832 100644 --- a/tests/pipelines/test_text_to_image_synthesis.py +++ b/tests/pipelines/test_text_to_image_synthesis.py @@ -15,7 +15,7 @@ class TextToImageSynthesisTest(unittest.TestCase): model_id = 'damo/cv_imagen_text-to-image-synthesis_tiny' test_text = {'text': '宇航员'} - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') def test_run_with_model_from_modelhub(self): model = Model.from_pretrained(self.model_id) pipe_line_text_to_image_synthesis = pipeline( @@ -24,7 +24,7 @@ class TextToImageSynthesisTest(unittest.TestCase): self.test_text)[OutputKeys.OUTPUT_IMG] print(np.sum(np.abs(img))) - @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') def test_run_with_model_name(self): pipe_line_text_to_image_synthesis = pipeline( task=Tasks.text_to_image_synthesis, model=self.model_id)