From 1336668f0e1e8c9975f485a4e2ecd3823c8b1b9a Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 11 Apr 2023 15:51:21 +0000 Subject: [PATCH] make style --- examples/community/composable_stable_diffusion.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/examples/community/composable_stable_diffusion.py b/examples/community/composable_stable_diffusion.py index b71a7f59c0..95292f5bda 100644 --- a/examples/community/composable_stable_diffusion.py +++ b/examples/community/composable_stable_diffusion.py @@ -22,6 +22,8 @@ from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer from diffusers import DiffusionPipeline from diffusers.configuration_utils import FrozenDict from diffusers.models import AutoencoderKL, UNet2DConditionModel +from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion import StableDiffusionPipelineOutput +from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker from diffusers.schedulers import ( DDIMScheduler, DPMSolverMultistepScheduler, @@ -30,10 +32,7 @@ from diffusers.schedulers import ( LMSDiscreteScheduler, PNDMScheduler, ) -from diffusers.utils import is_accelerate_available, deprecate, logging - -from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion import StableDiffusionPipelineOutput -from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker +from diffusers.utils import deprecate, is_accelerate_available, logging logger = logging.get_logger(__name__) # pylint: disable=invalid-name @@ -579,4 +578,3 @@ class ComposableStableDiffusionPipeline(DiffusionPipeline): return (image, has_nsfw_concept) return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept) -