Skip to content
Merged
Show file tree
Hide file tree
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
2 changes: 1 addition & 1 deletion src/diffusers/pipelines/qwenimage/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,8 @@
_import_structure["pipeline_qwenimage_controlnet"] = ["QwenImageControlNetPipeline"]
_import_structure["pipeline_qwenimage_controlnet_inpaint"] = ["QwenImageControlNetInpaintPipeline"]
_import_structure["pipeline_qwenimage_edit"] = ["QwenImageEditPipeline"]
_import_structure["pipeline_qwenimage_edit_plus"] = ["QwenImageEditPlusPipeline"]
_import_structure["pipeline_qwenimage_edit_inpaint"] = ["QwenImageEditInpaintPipeline"]
_import_structure["pipeline_qwenimage_edit_plus"] = ["QwenImageEditPlusPipeline"]
_import_structure["pipeline_qwenimage_img2img"] = ["QwenImageImg2ImgPipeline"]
_import_structure["pipeline_qwenimage_inpaint"] = ["QwenImageInpaintPipeline"]

Expand Down
5 changes: 5 additions & 0 deletions tests/pipelines/kandinsky/test_kandinsky.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,13 @@
import unittest

import numpy as np
import pytest
import torch
from transformers import XLMRobertaTokenizerFast

from diffusers import DDIMScheduler, KandinskyPipeline, KandinskyPriorPipeline, UNet2DConditionModel, VQModel
from diffusers.pipelines.kandinsky.text_encoder import MCLIPConfig, MultilingualCLIP
from diffusers.utils import is_transformers_version

from ...testing_utils import (
backend_empty_cache,
Expand Down Expand Up @@ -215,6 +217,9 @@ def get_dummy_inputs(self, device, seed=0):
dummy = Dummies()
return dummy.get_dummy_inputs(device=device, seed=seed)

@pytest.mark.xfail(
condition=is_transformers_version(">=", "4.56.2"), reason="Latest transformers changes the slices", strict=True
)
def test_kandinsky(self):
device = "cpu"

Expand Down
11 changes: 11 additions & 0 deletions tests/pipelines/kandinsky/test_kandinsky_combined.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,10 @@
import unittest

import numpy as np
import pytest

from diffusers import KandinskyCombinedPipeline, KandinskyImg2ImgCombinedPipeline, KandinskyInpaintCombinedPipeline
from diffusers.utils import is_transformers_version

from ...testing_utils import enable_full_determinism, require_torch_accelerator, torch_device
from ..test_pipelines_common import PipelineTesterMixin
Expand Down Expand Up @@ -73,6 +75,9 @@ def get_dummy_inputs(self, device, seed=0):
)
return inputs

@pytest.mark.xfail(
condition=is_transformers_version(">=", "4.56.2"), reason="Latest transformers changes the slices", strict=True
)
def test_kandinsky(self):
device = "cpu"

Expand Down Expand Up @@ -181,6 +186,9 @@ def get_dummy_inputs(self, device, seed=0):
inputs.pop("negative_image_embeds")
return inputs

@pytest.mark.xfail(
condition=is_transformers_version(">=", "4.56.2"), reason="Latest transformers changes the slices", strict=True
)
def test_kandinsky(self):
device = "cpu"

Expand Down Expand Up @@ -292,6 +300,9 @@ def get_dummy_inputs(self, device, seed=0):
inputs.pop("negative_image_embeds")
return inputs

@pytest.mark.xfail(
condition=is_transformers_version(">=", "4.56.2"), reason="Latest transformers changes the slices", strict=True
)
def test_kandinsky(self):
device = "cpu"

Expand Down
5 changes: 5 additions & 0 deletions tests/pipelines/kandinsky/test_kandinsky_img2img.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
import unittest

import numpy as np
import pytest
import torch
from PIL import Image
from transformers import XLMRobertaTokenizerFast
Expand All @@ -31,6 +32,7 @@
VQModel,
)
from diffusers.pipelines.kandinsky.text_encoder import MCLIPConfig, MultilingualCLIP
from diffusers.utils import is_transformers_version

from ...testing_utils import (
backend_empty_cache,
Expand Down Expand Up @@ -237,6 +239,9 @@ def get_dummy_inputs(self, device, seed=0):
dummies = Dummies()
return dummies.get_dummy_inputs(device=device, seed=seed)

@pytest.mark.xfail(
condition=is_transformers_version(">=", "4.56.2"), reason="Latest transformers changes the slices", strict=True
)
def test_kandinsky_img2img(self):
device = "cpu"

Expand Down
5 changes: 5 additions & 0 deletions tests/pipelines/kandinsky/test_kandinsky_inpaint.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,14 @@
import unittest

import numpy as np
import pytest
import torch
from PIL import Image
from transformers import XLMRobertaTokenizerFast

from diffusers import DDIMScheduler, KandinskyInpaintPipeline, KandinskyPriorPipeline, UNet2DConditionModel, VQModel
from diffusers.pipelines.kandinsky.text_encoder import MCLIPConfig, MultilingualCLIP
from diffusers.utils import is_transformers_version

from ...testing_utils import (
backend_empty_cache,
Expand Down Expand Up @@ -231,6 +233,9 @@ def get_dummy_inputs(self, device, seed=0):
dummies = Dummies()
return dummies.get_dummy_inputs(device=device, seed=seed)

@pytest.mark.xfail(
condition=is_transformers_version(">=", "4.56.2"), reason="Latest transformers changes the slices", strict=True
)
def test_kandinsky_inpaint(self):
device = "cpu"

Expand Down
Loading