Skip to content

Commit 35cc66d

Browse files
IrohXuXu CaoIrohCaoyiyixuxu
authored
Add pipeline_stable_diffusion_3_inpaint.py for SD3 Inference (#8709)
* Add pipeline_stable_diffusion_3_inpaint --------- Co-authored-by: Xu Cao <[email protected]> Co-authored-by: IrohXu <[email protected]> Co-authored-by: YiYi Xu <[email protected]>
1 parent 57084da commit 35cc66d

File tree

6 files changed

+1401
-2
lines changed

6 files changed

+1401
-2
lines changed

src/diffusers/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -288,6 +288,7 @@
288288
"StableCascadePriorPipeline",
289289
"StableDiffusion3ControlNetPipeline",
290290
"StableDiffusion3Img2ImgPipeline",
291+
"StableDiffusion3InpaintPipeline",
291292
"StableDiffusion3Pipeline",
292293
"StableDiffusionAdapterPipeline",
293294
"StableDiffusionAttendAndExcitePipeline",
@@ -690,6 +691,7 @@
690691
StableCascadePriorPipeline,
691692
StableDiffusion3ControlNetPipeline,
692693
StableDiffusion3Img2ImgPipeline,
694+
StableDiffusion3InpaintPipeline,
693695
StableDiffusion3Pipeline,
694696
StableDiffusionAdapterPipeline,
695697
StableDiffusionAttendAndExcitePipeline,

src/diffusers/pipelines/__init__.py

+10-2
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,11 @@
243243
"StableDiffusionLDM3DPipeline",
244244
]
245245
)
246-
_import_structure["stable_diffusion_3"] = ["StableDiffusion3Pipeline", "StableDiffusion3Img2ImgPipeline"]
246+
_import_structure["stable_diffusion_3"] = [
247+
"StableDiffusion3Pipeline",
248+
"StableDiffusion3Img2ImgPipeline",
249+
"StableDiffusion3InpaintPipeline",
250+
]
247251
_import_structure["stable_diffusion_attend_and_excite"] = ["StableDiffusionAttendAndExcitePipeline"]
248252
_import_structure["stable_diffusion_safe"] = ["StableDiffusionPipelineSafe"]
249253
_import_structure["stable_diffusion_sag"] = ["StableDiffusionSAGPipeline"]
@@ -523,7 +527,11 @@
523527
StableUnCLIPImg2ImgPipeline,
524528
StableUnCLIPPipeline,
525529
)
526-
from .stable_diffusion_3 import StableDiffusion3Img2ImgPipeline, StableDiffusion3Pipeline
530+
from .stable_diffusion_3 import (
531+
StableDiffusion3Img2ImgPipeline,
532+
StableDiffusion3InpaintPipeline,
533+
StableDiffusion3Pipeline,
534+
)
527535
from .stable_diffusion_attend_and_excite import StableDiffusionAttendAndExcitePipeline
528536
from .stable_diffusion_diffedit import StableDiffusionDiffEditPipeline
529537
from .stable_diffusion_gligen import StableDiffusionGLIGENPipeline, StableDiffusionGLIGENTextImagePipeline

src/diffusers/pipelines/stable_diffusion_3/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
else:
2626
_import_structure["pipeline_stable_diffusion_3"] = ["StableDiffusion3Pipeline"]
2727
_import_structure["pipeline_stable_diffusion_3_img2img"] = ["StableDiffusion3Img2ImgPipeline"]
28+
_import_structure["pipeline_stable_diffusion_3_inpaint"] = ["StableDiffusion3InpaintPipeline"]
2829

2930
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
3031
try:
@@ -35,6 +36,7 @@
3536
else:
3637
from .pipeline_stable_diffusion_3 import StableDiffusion3Pipeline
3738
from .pipeline_stable_diffusion_3_img2img import StableDiffusion3Img2ImgPipeline
39+
from .pipeline_stable_diffusion_3_inpaint import StableDiffusion3InpaintPipeline
3840

3941
else:
4042
import sys

0 commit comments

Comments
 (0)