From 6e1679557f60639ef672789bf37e65a2670e0f9b Mon Sep 17 00:00:00 2001 From: Amir Kasaei Date: Mon, 19 Aug 2024 12:38:44 +0330 Subject: [PATCH] Update pipeline_composable_stable_diffusion.py change 'from diffusers.pipeline_util import DiffusionPipeline' to 'from diffusers import DiffusionPipeline' due to updates in diffusers --- .../pipeline_composable_stable_diffusion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py b/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py index ba81083..e1304e0 100644 --- a/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py +++ b/composable_diffusion/composable_stable_diffusion/pipeline_composable_stable_diffusion.py @@ -11,7 +11,7 @@ from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer from packaging import version from diffusers.models import AutoencoderKL, UNet2DConditionModel -from diffusers.pipeline_utils import DiffusionPipeline +from diffusers import DiffusionPipeline from diffusers.schedulers import ( DDIMScheduler, DPMSolverMultistepScheduler, @@ -560,4 +560,4 @@ def __call__( if not return_dict: return (image, has_nsfw_concept) - return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept) \ No newline at end of file + return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)