Yingda Chen
3 years ago
1 changed files with 2 additions and 2 deletions
Unified View
Diff Options
+ 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' | model_id = 'damo/cv_imagen_text-to-image-synthesis_tiny' | ||||
| test_text = {'text': '宇航员'} | 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): | def test_run_with_model_from_modelhub(self): | ||||
| model = Model.from_pretrained(self.model_id) | model = Model.from_pretrained(self.model_id) | ||||
| pipe_line_text_to_image_synthesis = pipeline( | pipe_line_text_to_image_synthesis = pipeline( | ||||
| @@ -24,7 +24,7 @@ class TextToImageSynthesisTest(unittest.TestCase): | |||||
| self.test_text)[OutputKeys.OUTPUT_IMG] | self.test_text)[OutputKeys.OUTPUT_IMG] | ||||
| print(np.sum(np.abs(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): | def test_run_with_model_name(self): | ||||
| pipe_line_text_to_image_synthesis = pipeline( | pipe_line_text_to_image_synthesis = pipeline( | ||||
| task=Tasks.text_to_image_synthesis, model=self.model_id) | task=Tasks.text_to_image_synthesis, model=self.model_id) | ||||