From cf43991981efbb45c5a95f8a90459ed66eec4f22 Mon Sep 17 00:00:00 2001 From: XDUWQ <1300964705@qq.com> Date: Thu, 31 Aug 2023 23:42:01 +0800 Subject: [PATCH] skip tests --- tests/export/test_export_stable_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/export/test_export_stable_diffusion.py b/tests/export/test_export_stable_diffusion.py index a2e20198..91a877da 100644 --- a/tests/export/test_export_stable_diffusion.py +++ b/tests/export/test_export_stable_diffusion.py @@ -20,7 +20,7 @@ class TestExportStableDiffusion(unittest.TestCase): os.makedirs(self.tmp_dir) self.model_id = 'AI-ModelScope/stable-diffusion-v1-5' - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') def test_export_stable_diffusion(self): model = Model.from_pretrained(self.model_id) Exporter.from_model(model).export_onnx(