Skip to content

Commit fbc9a73

Browse files
authored
mps: skip unstable test (#3037)
1 parent 67c3518 commit fbc9a73

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

tests/pipelines/stable_unclip/test_stable_unclip_img2img.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,15 @@
1717
from diffusers.pipelines.pipeline_utils import DiffusionPipeline
1818
from diffusers.pipelines.stable_diffusion.stable_unclip_image_normalizer import StableUnCLIPImageNormalizer
1919
from diffusers.utils.import_utils import is_xformers_available
20-
from diffusers.utils.testing_utils import floats_tensor, load_image, load_numpy, require_torch_gpu, slow, torch_device
20+
from diffusers.utils.testing_utils import (
21+
floats_tensor,
22+
load_image,
23+
load_numpy,
24+
require_torch_gpu,
25+
skip_mps,
26+
slow,
27+
torch_device,
28+
)
2129

2230
from ...pipeline_params import TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS, TEXT_GUIDED_IMAGE_VARIATION_PARAMS
2331
from ...test_pipelines_common import (
@@ -147,6 +155,7 @@ def get_dummy_inputs(self, device, seed=0, pil_image=True):
147155
"output_type": "np",
148156
}
149157

158+
@skip_mps
150159
def test_image_embeds_none(self):
151160
device = "cpu" # ensure determinism for the device-dependent torch.Generator
152161
components = self.get_dummy_components()

0 commit comments

Comments
 (0)