Skip to content

GitLab

Commit ea90bf2b authored by anton-'s avatar anton-
Browse files

skip mps

parent 8cecc66a
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
tests/test_pipelines_common.py
View file @ ea90bf2b
......@@ -16,6 +16,7 @@ from diffusers import (
CycleDiffusionPipeline,
DanceDiffusionPipeline,
DiffusionPipeline,
RePaintPipeline,
StableDiffusionDepth2ImgPipeline,
StableDiffusionImg2ImgPipeline,
)
......@@ -72,6 +73,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline,
CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline,
):
# FIXME: inconsistent outputs on MPS
......@@ -178,6 +180,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline,
CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline,
):
# FIXME: inconsistent outputs on MPS
......@@ -292,6 +295,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline,
CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline,
):
# FIXME: inconsistent outputs on MPS
......@@ -358,6 +362,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline,
CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline,
StableDiffusionDepth2ImgPipeline,
):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment