diff --git a/tests/pipelines/test_fill_mask.py b/tests/pipelines/test_fill_mask.py index 450ada15..8c04eea5 100644 --- a/tests/pipelines/test_fill_mask.py +++ b/tests/pipelines/test_fill_mask.py @@ -122,7 +122,7 @@ class FillMaskTest(unittest.TestCase): model.model_dir, first_sequence='sentence', second_sequence=None) pipeline_ins = pipeline( Tasks.fill_mask, model=model, preprocessor=preprocessor) - for language in ['zh', 'en']: + for language in ['en']: ori_text = self.ori_texts[language] test_input = self.test_inputs[language].replace('[MASK]', '') with self.regress_tool.monitor_module_single_forward( diff --git a/tests/trainers/test_lora_diffusion_xl_trainer.py b/tests/trainers/test_lora_diffusion_xl_trainer.py index 3d415bfb..024d5139 100644 --- a/tests/trainers/test_lora_diffusion_xl_trainer.py +++ b/tests/trainers/test_lora_diffusion_xl_trainer.py @@ -68,7 +68,8 @@ class TestLoraDiffusionXLTrainer(unittest.TestCase): results_files = os.listdir(self.tmp_dir) self.assertIn(f'{trainer.timestamp}.log.json', results_files) - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + # need diffusers==0.24.0, skip in ci + @unittest.skip def test_lora_diffusion_xl_eval(self): model_id = 'AI-ModelScope/stable-diffusion-xl-base-1.0' model_revision = 'v1.0.2'