fix

huggingface · Oct 25, 2023 · 683b39c · 683b39c
1 parent 824fc57
commit 683b39c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/onnxruntime/test_stable_diffusion_pipeline.py b/tests/onnxruntime/test_stable_diffusion_pipeline.py
@@ -490,7 +490,6 @@ def test_vae_image_processor_pil(self):
                 self.assertTrue(np.allclose(in_np, out_np, atol=1e-6))
 
 
-@unittest.skipIf(parse(_diffusers_version) <= Version("0.21.4"), "not supported with this diffusers version")
 class ORTLatentConsistencyModelPipelineTest(ORTModelTestMixin):
     SUPPORTED_ARCHITECTURES = [
         "latent-consistency",
@@ -500,6 +499,7 @@ class ORTLatentConsistencyModelPipelineTest(ORTModelTestMixin):
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     @require_diffusers
+    @unittest.skipIf(parse(_diffusers_version) <= Version("0.21.4"), "not supported with this diffusers version")
     def test_compare_to_diffusers(self, model_arch: str):
         ort_pipeline = self.ORTMODEL_CLASS.from_pretrained(MODEL_NAMES[model_arch], export=True)
         self.assertIsInstance(ort_pipeline.text_encoder, ORTModelTextEncoder)