mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
Skip mps in text-to-video tests (#2792)
* Skip mps in text-to-video tests. * style * Skip UNet3D mps tests.
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user