skip tests

This commit is contained in:
XDUWQ
2023-08-31 23:42:01 +08:00
parent c89de1a825
commit cf43991981

View File

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