Unverified Commit aa0531fa authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Skip `mps` in text-to-video tests (#2792)

* Skip mps in text-to-video tests.

* style

* Skip UNet3D mps tests.
parent dc5b4e23
......@@ -23,6 +23,7 @@ from diffusers.models.attention_processor import LoRAAttnProcessor
from diffusers.utils import (
floats_tensor,
logging,
skip_mps,
torch_device,
)
from diffusers.utils.import_utils import is_xformers_available
......@@ -60,6 +61,7 @@ def create_lora_layers(model):
return lora_attn_procs
@skip_mps
class UNet3DConditionModelTests(ModelTesterMixin, unittest.TestCase):
model_class = UNet3DConditionModel
......
......@@ -35,6 +35,7 @@ from ...test_pipelines_common import PipelineTesterMixin
torch.backends.cuda.matmul.allow_tf32 = False
@skip_mps
class TextToVideoSDPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
pipeline_class = TextToVideoSDPipeline
params = TEXT_TO_IMAGE_PARAMS
......@@ -155,12 +156,12 @@ class TextToVideoSDPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
def test_num_images_per_prompt(self):
pass
@skip_mps
def test_progress_bar(self):
return super().test_progress_bar()
@slow
@skip_mps
class TextToVideoSDPipelineSlowTests(unittest.TestCase):
def test_full_model(self):
expected_video = load_numpy(
......
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