Skip to content

Commit 33fafe3

Browse files
juancopi81yiyixuxu
andauthored
Add PAG support to StableDiffusionControlNetPAGInpaintPipeline (#8875)
* Add pag to controlnet inpainting pipeline --------- Co-authored-by: YiYi Xu <yixu310@gmail.com>
1 parent c4a8979 commit 33fafe3

File tree

9 files changed

+1816
-1
lines changed

9 files changed

+1816
-1
lines changed

docs/source/en/api/pipelines/pag.md

+3
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,9 @@ Since RegEx is supported as a way for matching layer identifiers, it is crucial
5555

5656
## StableDiffusionControlNetPAGPipeline
5757
[[autodoc]] StableDiffusionControlNetPAGPipeline
58+
59+
## StableDiffusionControlNetPAGInpaintPipeline
60+
[[autodoc]] StableDiffusionControlNetPAGInpaintPipeline
5861
- all
5962
- __call__
6063

src/diffusers/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -328,6 +328,7 @@
328328
"StableDiffusionAttendAndExcitePipeline",
329329
"StableDiffusionControlNetImg2ImgPipeline",
330330
"StableDiffusionControlNetInpaintPipeline",
331+
"StableDiffusionControlNetPAGInpaintPipeline",
331332
"StableDiffusionControlNetPAGPipeline",
332333
"StableDiffusionControlNetPipeline",
333334
"StableDiffusionControlNetXSPipeline",
@@ -778,6 +779,7 @@
778779
StableDiffusionAttendAndExcitePipeline,
779780
StableDiffusionControlNetImg2ImgPipeline,
780781
StableDiffusionControlNetInpaintPipeline,
782+
StableDiffusionControlNetPAGInpaintPipeline,
781783
StableDiffusionControlNetPAGPipeline,
782784
StableDiffusionControlNetPipeline,
783785
StableDiffusionControlNetXSPipeline,

src/diffusers/pipelines/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@
158158
)
159159
_import_structure["pag"].extend(
160160
[
161+
"StableDiffusionControlNetPAGInpaintPipeline",
161162
"AnimateDiffPAGPipeline",
162163
"KolorsPAGPipeline",
163164
"HunyuanDiTPAGPipeline",
@@ -566,6 +567,7 @@
566567
KolorsPAGPipeline,
567568
PixArtSigmaPAGPipeline,
568569
StableDiffusion3PAGPipeline,
570+
StableDiffusionControlNetPAGInpaintPipeline,
569571
StableDiffusionControlNetPAGPipeline,
570572
StableDiffusionPAGPipeline,
571573
StableDiffusionXLControlNetPAGImg2ImgPipeline,

src/diffusers/pipelines/auto_pipeline.py

+2
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@
6161
HunyuanDiTPAGPipeline,
6262
PixArtSigmaPAGPipeline,
6363
StableDiffusion3PAGPipeline,
64+
StableDiffusionControlNetPAGInpaintPipeline,
6465
StableDiffusionControlNetPAGPipeline,
6566
StableDiffusionPAGPipeline,
6667
StableDiffusionXLControlNetPAGImg2ImgPipeline,
@@ -148,6 +149,7 @@
148149
("kandinsky", KandinskyInpaintCombinedPipeline),
149150
("kandinsky22", KandinskyV22InpaintCombinedPipeline),
150151
("stable-diffusion-controlnet", StableDiffusionControlNetInpaintPipeline),
152+
("stable-diffusion-controlnet-pag", StableDiffusionControlNetPAGInpaintPipeline),
151153
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetInpaintPipeline),
152154
("stable-diffusion-xl-pag", StableDiffusionXLPAGInpaintPipeline),
153155
("flux", FluxInpaintPipeline),

src/diffusers/pipelines/pag/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
2424
else:
2525
_import_structure["pipeline_pag_controlnet_sd"] = ["StableDiffusionControlNetPAGPipeline"]
26+
_import_structure["pipeline_pag_controlnet_sd_inpaint"] = ["StableDiffusionControlNetPAGInpaintPipeline"]
2627
_import_structure["pipeline_pag_controlnet_sd_xl"] = ["StableDiffusionXLControlNetPAGPipeline"]
2728
_import_structure["pipeline_pag_controlnet_sd_xl_img2img"] = ["StableDiffusionXLControlNetPAGImg2ImgPipeline"]
2829
_import_structure["pipeline_pag_hunyuandit"] = ["HunyuanDiTPAGPipeline"]
@@ -44,6 +45,7 @@
4445
from ...utils.dummy_torch_and_transformers_objects import *
4546
else:
4647
from .pipeline_pag_controlnet_sd import StableDiffusionControlNetPAGPipeline
48+
from .pipeline_pag_controlnet_sd_inpaint import StableDiffusionControlNetPAGInpaintPipeline
4749
from .pipeline_pag_controlnet_sd_xl import StableDiffusionXLControlNetPAGPipeline
4850
from .pipeline_pag_controlnet_sd_xl_img2img import StableDiffusionXLControlNetPAGImg2ImgPipeline
4951
from .pipeline_pag_hunyuandit import HunyuanDiTPAGPipeline

0 commit comments

Comments
 (0)