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 ( ...@@ -16,6 +16,7 @@ from diffusers import (
CycleDiffusionPipeline, CycleDiffusionPipeline,
DanceDiffusionPipeline, DanceDiffusionPipeline,
DiffusionPipeline, DiffusionPipeline,
RePaintPipeline,
StableDiffusionDepth2ImgPipeline, StableDiffusionDepth2ImgPipeline,
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
) )
...@@ -72,6 +73,7 @@ class PipelineTesterMixin: ...@@ -72,6 +73,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in ( if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline, DanceDiffusionPipeline,
CycleDiffusionPipeline, CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
): ):
# FIXME: inconsistent outputs on MPS # FIXME: inconsistent outputs on MPS
...@@ -178,6 +180,7 @@ class PipelineTesterMixin: ...@@ -178,6 +180,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in ( if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline, DanceDiffusionPipeline,
CycleDiffusionPipeline, CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
): ):
# FIXME: inconsistent outputs on MPS # FIXME: inconsistent outputs on MPS
...@@ -292,6 +295,7 @@ class PipelineTesterMixin: ...@@ -292,6 +295,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in ( if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline, DanceDiffusionPipeline,
CycleDiffusionPipeline, CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
): ):
# FIXME: inconsistent outputs on MPS # FIXME: inconsistent outputs on MPS
...@@ -358,6 +362,7 @@ class PipelineTesterMixin: ...@@ -358,6 +362,7 @@ class PipelineTesterMixin:
if torch_device == "mps" and self.pipeline_class in ( if torch_device == "mps" and self.pipeline_class in (
DanceDiffusionPipeline, DanceDiffusionPipeline,
CycleDiffusionPipeline, CycleDiffusionPipeline,
RePaintPipeline,
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
StableDiffusionDepth2ImgPipeline, 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