add PAG support for SD Controlnet Img2Img

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

@@ -44,3 +44,8 @@ The abstract from the paper is:
 [[autodoc]] StableDiffusionXLControlNetPAGPipeline
 	- all
 	- __call__
+
+## StableDiffusionControlNetPAGImg2ImgPipeline
+[[autodoc]] StableDiffusionControlNetPAGImg2ImgPipeline
+	- all
+	- __call__

src/diffusers/__init__.py

@@ -289,6 +289,7 @@
             "StableDiffusionAdapterPipeline",
             "StableDiffusionAttendAndExcitePipeline",
             "StableDiffusionControlNetImg2ImgPipeline",
+            "StableDiffusionControlNetPAGImg2ImgPipeline",
             "StableDiffusionControlNetInpaintPipeline",
             "StableDiffusionControlNetPipeline",
             "StableDiffusionControlNetXSPipeline",
@@ -697,6 +698,7 @@
             StableDiffusionGLIGENTextImagePipeline,
             StableDiffusionImageVariationPipeline,
             StableDiffusionImg2ImgPipeline,
+            StableDiffusionControlNetPAGImg2ImgPipeline,
             StableDiffusionInpaintPipeline,
             StableDiffusionInpaintPipelineLegacy,
             StableDiffusionInstructPix2PixPipeline,

src/diffusers/pipelines/__init__.py

@@ -145,6 +145,7 @@
             "StableDiffusionXLPAGPipeline",
             "StableDiffusionXLPAGInpaintPipeline",
             "StableDiffusionXLControlNetPAGPipeline",
+            "StableDiffusionControlNetPAGImg2ImgPipeline",
             "StableDiffusionXLPAGImg2ImgPipeline",
         ]
     )
@@ -497,6 +498,7 @@
             StableDiffusionXLPAGImg2ImgPipeline,
             StableDiffusionXLPAGInpaintPipeline,
             StableDiffusionXLPAGPipeline,
+            StableDiffusionControlNetPAGImg2ImgPipeline,
         )
         from .paint_by_example import PaintByExamplePipeline
         from .pia import PIAPipeline

src/diffusers/pipelines/auto_pipeline.py

@@ -52,6 +52,7 @@
     StableDiffusionXLPAGImg2ImgPipeline,
     StableDiffusionXLPAGInpaintPipeline,
     StableDiffusionXLPAGPipeline,
+    StableDiffusionControlNetPAGImg2ImgPipeline,
 )
 from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
 from .stable_cascade import StableCascadeCombinedPipeline, StableCascadeDecoderPipeline
@@ -92,6 +93,7 @@
         ("stable-diffusion-pag", StableDiffusionPAGPipeline),
         ("stable-diffusion-xl-pag", StableDiffusionXLPAGPipeline),
         ("stable-diffusion-xl-controlnet-pag", StableDiffusionXLControlNetPAGPipeline),
+        ("stable-diffusion-controlnet-pag-img2img", StableDiffusionControlNetPAGImg2ImgPipeline),
     ]
 )
 

src/diffusers/pipelines/pag/__init__.py

@@ -27,6 +27,7 @@
     _import_structure["pipeline_pag_sd_xl"] = ["StableDiffusionXLPAGPipeline"]
     _import_structure["pipeline_pag_sd_xl_img2img"] = ["StableDiffusionXLPAGImg2ImgPipeline"]
     _import_structure["pipeline_pag_sd_xl_inpaint"] = ["StableDiffusionXLPAGInpaintPipeline"]
+    _import_structure["pipeline_pag_controlnet_sd_img2img"] = ["StableDiffusionControlNetPAGImg2ImgPipeline"]
 
 if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
     try:
@@ -41,6 +42,7 @@
         from .pipeline_pag_sd_xl import StableDiffusionXLPAGPipeline
         from .pipeline_pag_sd_xl_img2img import StableDiffusionXLPAGImg2ImgPipeline
         from .pipeline_pag_sd_xl_inpaint import StableDiffusionXLPAGInpaintPipeline
+        from .pipeline_pag_controlnet_sd_img2img import StableDiffusionControlNetPAGImg2ImgPipeline
 
 else:
     import sys