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 (
LMSDiscreteScheduler,
PNDMScheduler,
)
from diffusers.utils import is_accelerate_available
from diffusers.utils import is_accelerate_available, deprecate, logging
from ...utils import deprecate, logging
from . import StableDiffusionPipelineOutput
from .safety_checker import StableDiffusionSafetyChecker
from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion import StableDiffusionPipelineOutput
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
......@@ -580,3 +579,4 @@ class ComposableStableDiffusionPipeline(DiffusionPipeline):
return (image, 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