File tree 1 file changed +3
-5
lines changed 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change 22
22
from diffusers import DiffusionPipeline
23
23
from diffusers .configuration_utils import FrozenDict
24
24
from 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
25
27
from diffusers .schedulers import (
26
28
DDIMScheduler ,
27
29
DPMSolverMultistepScheduler ,
30
32
LMSDiscreteScheduler ,
31
33
PNDMScheduler ,
32
34
)
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
37
36
38
37
39
38
logger = logging .get_logger (__name__ ) # pylint: disable=invalid-name
@@ -579,4 +578,3 @@ def __call__(
579
578
return (image , has_nsfw_concept )
580
579
581
580
return StableDiffusionPipelineOutput (images = image , nsfw_content_detected = has_nsfw_concept )
582
-
You can’t perform that action at this time.
0 commit comments