Unverified Commit 881a6b58 authored by J N Hearns's avatar J N Hearns Committed by GitHub
Browse files

Fix imports for composable_stable_diffusion pipeline (#3002)

* Update composable_stable_diffusion.py

Fix imports

* Formatting

* Formatting

* Formatting
parent cb9d77af
...@@ -30,11 +30,10 @@ from diffusers.schedulers import ( ...@@ -30,11 +30,10 @@ from diffusers.schedulers import (
LMSDiscreteScheduler, LMSDiscreteScheduler,
PNDMScheduler, PNDMScheduler,
) )
from diffusers.utils import is_accelerate_available from diffusers.utils import is_accelerate_available, deprecate, logging
from ...utils import deprecate, logging from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion import StableDiffusionPipelineOutput
from . import StableDiffusionPipelineOutput from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
from .safety_checker import StableDiffusionSafetyChecker
logger = logging.get_logger(__name__) # pylint: disable=invalid-name logger = logging.get_logger(__name__) # pylint: disable=invalid-name
...@@ -580,3 +579,4 @@ class ComposableStableDiffusionPipeline(DiffusionPipeline): ...@@ -580,3 +579,4 @@ class ComposableStableDiffusionPipeline(DiffusionPipeline):
return (image, has_nsfw_concept) return (image, has_nsfw_concept)
return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept) return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment