Skip to content

Add pipeline_stable_diffusion_3_inpaint.py for SD3 Inference #8709

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 17 commits into from
Jul 9, 2024
Merged

Add pipeline_stable_diffusion_3_inpaint.py for SD3 Inference #8709

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
02c0b11
Add pipeline_stable_diffusion_3_inpaint
Jun 26, 2024
9419761
Merge branch 'main' into main
IrohXu Jun 26, 2024
76d775c
fix bugs
Jun 27, 2024
b5b1550
fix bugs
Jun 27, 2024
7598fbb
fix bugs based on comments
IrohCao Jun 27, 2024
79b1588
fix bugs by using SDXL and SD pipeline
IrohCao Jul 1, 2024
52e67f0
fix bugs by using SDXL and SD pipeline
IrohCao Jul 1, 2024
ec92e7e
fix bugs by using SDXL and SD pipeline
IrohCao Jul 1, 2024
7f8c8cd
fix bugs by using SDXL and SD pipeline
IrohCao Jul 1, 2024
169b32d
improve SD3-inpainting
IrohCao Jul 1, 2024
db67af1
update test_pipeline_stable_diffusion_3_inpaint
IrohCao Jul 1, 2024
64f02ae
update test_pipeline_stable_diffusion_3_inpaint
IrohCao Jul 1, 2024
3f382a4
fix bugs in pipeline_stable_diffusion_3_inpaint.py
IrohCao Jul 3, 2024
86852f2
fix structure bugs
IrohCao Jul 3, 2024
9ec0e37
update with make fix-copies
IrohCao Jul 3, 2024
269415c
Merge branch 'main' into main
yiyixuxu Jul 8, 2024
e3aa7b5
update num_images_per_prompt
IrohCao Jul 8, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions src/diffusers/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -288,6 +288,7 @@
"StableCascadePriorPipeline",
"StableDiffusion3ControlNetPipeline",
"StableDiffusion3Img2ImgPipeline",
"StableDiffusion3InpaintPipeline",
"StableDiffusion3Pipeline",
"StableDiffusionAdapterPipeline",
"StableDiffusionAttendAndExcitePipeline",
Expand Down Expand Up @@ -690,6 +691,7 @@
StableCascadePriorPipeline,
StableDiffusion3ControlNetPipeline,
StableDiffusion3Img2ImgPipeline,
StableDiffusion3InpaintPipeline,
StableDiffusion3Pipeline,
StableDiffusionAdapterPipeline,
StableDiffusionAttendAndExcitePipeline,
Expand Down
12 changes: 10 additions & 2 deletions src/diffusers/pipelines/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -243,7 +243,11 @@
"StableDiffusionLDM3DPipeline",
]
)
_import_structure["stable_diffusion_3"] = ["StableDiffusion3Pipeline", "StableDiffusion3Img2ImgPipeline"]
_import_structure["stable_diffusion_3"] = [
"StableDiffusion3Pipeline",
"StableDiffusion3Img2ImgPipeline",
"StableDiffusion3InpaintPipeline",
]
_import_structure["stable_diffusion_attend_and_excite"] = ["StableDiffusionAttendAndExcitePipeline"]
_import_structure["stable_diffusion_safe"] = ["StableDiffusionPipelineSafe"]
_import_structure["stable_diffusion_sag"] = ["StableDiffusionSAGPipeline"]
Expand Down Expand Up @@ -523,7 +527,11 @@
StableUnCLIPImg2ImgPipeline,
StableUnCLIPPipeline,
)
from .stable_diffusion_3 import StableDiffusion3Img2ImgPipeline, StableDiffusion3Pipeline
from .stable_diffusion_3 import (
StableDiffusion3Img2ImgPipeline,
StableDiffusion3InpaintPipeline,
StableDiffusion3Pipeline,
)
from .stable_diffusion_attend_and_excite import StableDiffusionAttendAndExcitePipeline
from .stable_diffusion_diffedit import StableDiffusionDiffEditPipeline
from .stable_diffusion_gligen import StableDiffusionGLIGENPipeline, StableDiffusionGLIGENTextImagePipeline
Expand Down
2 changes: 2 additions & 0 deletions src/diffusers/pipelines/stable_diffusion_3/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
else:
_import_structure["pipeline_stable_diffusion_3"] = ["StableDiffusion3Pipeline"]
_import_structure["pipeline_stable_diffusion_3_img2img"] = ["StableDiffusion3Img2ImgPipeline"]
_import_structure["pipeline_stable_diffusion_3_inpaint"] = ["StableDiffusion3InpaintPipeline"]

if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
try:
Expand All @@ -35,6 +36,7 @@
else:
from .pipeline_stable_diffusion_3 import StableDiffusion3Pipeline
from .pipeline_stable_diffusion_3_img2img import StableDiffusion3Img2ImgPipeline
from .pipeline_stable_diffusion_3_inpaint import StableDiffusion3InpaintPipeline

else:
import sys
Expand Down
Loading