mirror of
https://github.com/modelscope/modelscope.git
synced 2026-07-10 04:22:33 +02:00
skip tests
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user