Browse Source

[to #42322933] do not run imagen on gated for now

master
Yingda Chen 3 years ago
parent
commit
059c8fd248
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/pipelines/test_text_to_image_synthesis.py

+ 2
- 2
tests/pipelines/test_text_to_image_synthesis.py View File

@@ -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)


Loading…
Cancel
Save