diff --git a/tests/openvino/test_stable_diffusion.py b/tests/openvino/test_stable_diffusion.py index 3c0c90475a..e04e2d6fd3 100644 --- a/tests/openvino/test_stable_diffusion.py +++ b/tests/openvino/test_stable_diffusion.py @@ -121,7 +121,6 @@ def callback_fn(step: int, timestep: int, latents: np.ndarray) -> None: self.assertEqual(callback_fn.number_of_steps, inputs["num_inference_steps"]) @parameterized.expand(SUPPORTED_ARCHITECTURES) - @require_diffusers def test_shape(self, model_arch: str): height, width, batch_size = 128, 64, 1 pipeline = self.MODEL_CLASS.from_pretrained(MODEL_NAMES[model_arch], export=True)