From c9b034b94e102164903564f471ef58774264e587 Mon Sep 17 00:00:00 2001 From: DN6 Date: Thu, 17 Apr 2025 13:50:14 +0530 Subject: [PATCH 1/2] update --- src/diffusers/pipelines/__init__.py | 2 -- src/diffusers/pipelines/deprecated/__init__.py | 2 ++ 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/__init__.py b/src/diffusers/pipelines/__init__.py index 6fdcdf1f53..c9700aee8f 100644 --- a/src/diffusers/pipelines/__init__.py +++ b/src/diffusers/pipelines/__init__.py @@ -508,7 +508,6 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: AudioLDM2UNet2DConditionModel, ) from .aura_flow import AuraFlowPipeline - from .blip_diffusion import BlipDiffusionPipeline from .cogvideo import ( CogVideoXFunControlPipeline, CogVideoXImageToVideoPipeline, @@ -518,7 +517,6 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: from .cogview3 import CogView3PlusPipeline from .cogview4 import CogView4ControlPipeline, CogView4Pipeline from .controlnet import ( - BlipDiffusionControlNetPipeline, StableDiffusionControlNetImg2ImgPipeline, StableDiffusionControlNetInpaintPipeline, StableDiffusionControlNetPipeline, diff --git a/src/diffusers/pipelines/deprecated/__init__.py b/src/diffusers/pipelines/deprecated/__init__.py index 7bc85375f7..9eb5fffee7 100644 --- a/src/diffusers/pipelines/deprecated/__init__.py +++ b/src/diffusers/pipelines/deprecated/__init__.py @@ -181,6 +181,8 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: else: from .alt_diffusion import AltDiffusionImg2ImgPipeline, AltDiffusionPipeline, AltDiffusionPipelineOutput from .audioldm import AudioLDMPipeline + from .blip_diffusion import BlipDiffusionPipeline + from .controlnet import BlipDiffusionControlNetPipeline from .controlnet_xs import StableDiffusionControlNetXSPipeline, StableDiffusionXLControlNetXSPipeline from .i2vgen_xl import I2VGenXLPipeline from .latte import LattePipeline From 01ad7116454a3cb1e290b0580a3b052b85922880 Mon Sep 17 00:00:00 2001 From: DN6 Date: Thu, 17 Apr 2025 14:04:56 +0530 Subject: [PATCH 2/2] update --- tests/pipelines/blipdiffusion/test_blipdiffusion.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/tests/pipelines/blipdiffusion/test_blipdiffusion.py b/tests/pipelines/blipdiffusion/test_blipdiffusion.py index db8d36b23a..10ff4ea8b9 100644 --- a/tests/pipelines/blipdiffusion/test_blipdiffusion.py +++ b/tests/pipelines/blipdiffusion/test_blipdiffusion.py @@ -23,10 +23,9 @@ from transformers.models.clip.configuration_clip import CLIPTextConfig from diffusers import AutoencoderKL, BlipDiffusionPipeline, PNDMScheduler, UNet2DConditionModel from diffusers.utils.testing_utils import enable_full_determinism -from src.diffusers.pipelines.blip_diffusion.blip_image_processing import BlipImageProcessor -from src.diffusers.pipelines.blip_diffusion.modeling_blip2 import Blip2QFormerModel -from src.diffusers.pipelines.blip_diffusion.modeling_ctx_clip import ContextCLIPTextModel - +from diffusers.pipelines.deprecated.blip_diffusion.blip_image_processing import BlipImageProcessor +from diffusers.pipelines.deprecated.blip_diffusion.modeling_blip2 import Blip2QFormerModel +from diffusers.pipelines.deprecated.blip_diffusion.modeling_ctx_clip import ContextCLIPTextModel from ..test_pipelines_common import PipelineTesterMixin @@ -195,9 +194,9 @@ class BlipDiffusionPipelineFastTests(PipelineTesterMixin, unittest.TestCase): [0.5329548, 0.8372512, 0.33269387, 0.82096875, 0.43657133, 0.3783, 0.5953028, 0.51934963, 0.42142007] ) - assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2, ( - f" expected_slice {image_slice.flatten()}, but got {image_slice.flatten()}" - ) + assert ( + np.abs(image_slice.flatten() - expected_slice).max() < 1e-2 + ), f" expected_slice {image_slice.flatten()}, but got {image_slice.flatten()}" @unittest.skip("Test not supported because of complexities in deriving query_embeds.") def test_encode_prompt_works_in_isolation(self):