Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions tests/pipelines/stable_cascade/test_stable_cascade_prior.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,13 @@
import unittest

import numpy as np
import pytest
import torch
from transformers import CLIPTextConfig, CLIPTextModelWithProjection, CLIPTokenizer

from diffusers import DDPMWuerstchenScheduler, StableCascadePriorPipeline
from diffusers.models import StableCascadeUNet
from diffusers.utils import is_transformers_version
from diffusers.utils.import_utils import is_peft_available

from ...testing_utils import (
Expand Down Expand Up @@ -154,6 +156,11 @@ def get_dummy_inputs(self, device, seed=0):
}
return inputs

@pytest.mark.xfail(
condition=is_transformers_version(">=", "4.57.1"),
reason="Test fails with the latest transformers version",
strict=False,
)
def test_wuerstchen_prior(self):
device = "cpu"

Expand Down
Loading