1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

Fix imports for composable_stable_diffusion pipeline (#3002)

* Update composable_stable_diffusion.py

Fix imports

* Formatting

* Formatting

* Formatting
This commit is contained in:
J N Hearns
2023-04-11 09:50:25 -06:00
committed by Daniel Gu
parent 531192367f
commit 1ee04a5e60

View File

@@ -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)