File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change 2222from diffusers import DiffusionPipeline
2323from diffusers .configuration_utils import FrozenDict
2424from diffusers .models import AutoencoderKL , UNet2DConditionModel
25+ from diffusers .pipelines .stable_diffusion .pipeline_stable_diffusion import StableDiffusionPipelineOutput
26+ from diffusers .pipelines .stable_diffusion .safety_checker import StableDiffusionSafetyChecker
2527from diffusers .schedulers import (
2628 DDIMScheduler ,
2729 DPMSolverMultistepScheduler ,
3032 LMSDiscreteScheduler ,
3133 PNDMScheduler ,
3234)
33- from diffusers .utils import is_accelerate_available , deprecate , logging
34-
35- from diffusers .pipelines .stable_diffusion .pipeline_stable_diffusion import StableDiffusionPipelineOutput
36- from diffusers .pipelines .stable_diffusion .safety_checker import StableDiffusionSafetyChecker
35+ from diffusers .utils import deprecate , is_accelerate_available , logging
3736
3837
3938logger = logging .get_logger (__name__ ) # pylint: disable=invalid-name
@@ -579,4 +578,3 @@ def __call__(
579578 return (image , has_nsfw_concept )
580579
581580 return StableDiffusionPipelineOutput (images = image , nsfw_content_detected = has_nsfw_concept )
582-
You can’t perform that action at this time.
0 commit comments