Skip to content

Commit 9003d75

Browse files
satani99yiyixuxu
andauthored
Add StableDiffusionXLControlNetPAGImg2ImgPipeline (#8990)
* Added pad controlnet sdxl img2img pipeline --------- Co-authored-by: YiYi Xu <[email protected]>
1 parent 750bd79 commit 9003d75

File tree

9 files changed

+1983
-1
lines changed

9 files changed

+1983
-1
lines changed

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,10 @@ Since RegEx is supported as a way for matching layer identifiers, it is crucial
7878
- all
7979
- __call__
8080

81+
## StableDiffusionXLControlNetPAGImg2ImgPipeline
82+
[[autodoc]] StableDiffusionXLControlNetPAGImg2ImgPipeline
83+
- all
84+
- __call__
8185

8286
## StableDiffusion3PAGPipeline
8387
[[autodoc]] StableDiffusion3PAGPipeline

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -346,6 +346,7 @@
346346
"StableDiffusionXLAdapterPipeline",
347347
"StableDiffusionXLControlNetImg2ImgPipeline",
348348
"StableDiffusionXLControlNetInpaintPipeline",
349+
"StableDiffusionXLControlNetPAGImg2ImgPipeline",
349350
"StableDiffusionXLControlNetPAGPipeline",
350351
"StableDiffusionXLControlNetPipeline",
351352
"StableDiffusionXLControlNetXSPipeline",
@@ -787,6 +788,7 @@
787788
StableDiffusionXLAdapterPipeline,
788789
StableDiffusionXLControlNetImg2ImgPipeline,
789790
StableDiffusionXLControlNetInpaintPipeline,
791+
StableDiffusionXLControlNetPAGImg2ImgPipeline,
790792
StableDiffusionXLControlNetPAGPipeline,
791793
StableDiffusionXLControlNetPipeline,
792794
StableDiffusionXLControlNetXSPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,7 @@
154154
"StableDiffusionControlNetPAGPipeline",
155155
"StableDiffusionXLPAGPipeline",
156156
"StableDiffusionXLPAGInpaintPipeline",
157+
"StableDiffusionXLControlNetPAGImg2ImgPipeline",
157158
"StableDiffusionXLControlNetPAGPipeline",
158159
"StableDiffusionXLPAGImg2ImgPipeline",
159160
"PixArtSigmaPAGPipeline",
@@ -547,6 +548,7 @@
547548
StableDiffusion3PAGPipeline,
548549
StableDiffusionControlNetPAGPipeline,
549550
StableDiffusionPAGPipeline,
551+
StableDiffusionXLControlNetPAGImg2ImgPipeline,
550552
StableDiffusionXLControlNetPAGPipeline,
551553
StableDiffusionXLPAGImg2ImgPipeline,
552554
StableDiffusionXLPAGInpaintPipeline,

src/diffusers/pipelines/auto_pipeline.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
StableDiffusion3PAGPipeline,
5757
StableDiffusionControlNetPAGPipeline,
5858
StableDiffusionPAGPipeline,
59+
StableDiffusionXLControlNetPAGImg2ImgPipeline,
5960
StableDiffusionXLControlNetPAGPipeline,
6061
StableDiffusionXLPAGImg2ImgPipeline,
6162
StableDiffusionXLPAGInpaintPipeline,
@@ -123,6 +124,7 @@
123124
("stable-diffusion-controlnet", StableDiffusionControlNetImg2ImgPipeline),
124125
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetImg2ImgPipeline),
125126
("stable-diffusion-xl-pag", StableDiffusionXLPAGImg2ImgPipeline),
127+
("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGImg2ImgPipeline),
126128
("lcm", LatentConsistencyModelImg2ImgPipeline),
127129
]
128130
)

src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1538,7 +1538,6 @@ def __call__(
15381538
if isinstance(controlnet_cond_scale, list):
15391539
controlnet_cond_scale = controlnet_cond_scale[0]
15401540
cond_scale = controlnet_cond_scale * controlnet_keep[i]
1541-
15421541
down_block_res_samples, mid_block_res_sample = self.controlnet(
15431542
control_model_input,
15441543
t,

src/diffusers/pipelines/pag/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
else:
2525
_import_structure["pipeline_pag_controlnet_sd"] = ["StableDiffusionControlNetPAGPipeline"]
2626
_import_structure["pipeline_pag_controlnet_sd_xl"] = ["StableDiffusionXLControlNetPAGPipeline"]
27+
_import_structure["pipeline_pag_controlnet_sd_xl_img2img"] = ["StableDiffusionXLControlNetPAGImg2ImgPipeline"]
2728
_import_structure["pipeline_pag_hunyuandit"] = ["HunyuanDiTPAGPipeline"]
2829
_import_structure["pipeline_pag_kolors"] = ["KolorsPAGPipeline"]
2930
_import_structure["pipeline_pag_pixart_sigma"] = ["PixArtSigmaPAGPipeline"]
@@ -44,6 +45,7 @@
4445
else:
4546
from .pipeline_pag_controlnet_sd import StableDiffusionControlNetPAGPipeline
4647
from .pipeline_pag_controlnet_sd_xl import StableDiffusionXLControlNetPAGPipeline
48+
from .pipeline_pag_controlnet_sd_xl_img2img import StableDiffusionXLControlNetPAGImg2ImgPipeline
4749
from .pipeline_pag_hunyuandit import HunyuanDiTPAGPipeline
4850
from .pipeline_pag_kolors import KolorsPAGPipeline
4951
from .pipeline_pag_pixart_sigma import PixArtSigmaPAGPipeline

0 commit comments

Comments
 (0)