Skip to content

Commit 4e36bb0

Browse files
haofanwanggithub-actions[bot]yiyixuxu
authored
Support ControlNet-Inpainting for Qwen-Image (#12301)
* add qwen-image-cn-inpaint --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: yiyixuxu <yixu310@gmail.com>
1 parent f50b18e commit 4e36bb0

File tree

5 files changed

+962
-0
lines changed

5 files changed

+962
-0
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -510,6 +510,7 @@
510510
"PixArtAlphaPipeline",
511511
"PixArtSigmaPAGPipeline",
512512
"PixArtSigmaPipeline",
513+
"QwenImageControlNetInpaintPipeline",
513514
"QwenImageControlNetPipeline",
514515
"QwenImageEditInpaintPipeline",
515516
"QwenImageEditPipeline",
@@ -1163,6 +1164,7 @@
11631164
PixArtAlphaPipeline,
11641165
PixArtSigmaPAGPipeline,
11651166
PixArtSigmaPipeline,
1167+
QwenImageControlNetInpaintPipeline,
11661168
QwenImageControlNetPipeline,
11671169
QwenImageEditInpaintPipeline,
11681170
QwenImageEditPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -394,6 +394,7 @@
394394
"QwenImageInpaintPipeline",
395395
"QwenImageEditPipeline",
396396
"QwenImageEditInpaintPipeline",
397+
"QwenImageControlNetInpaintPipeline",
397398
"QwenImageControlNetPipeline",
398399
]
399400
try:
@@ -714,6 +715,7 @@
714715
from .pia import PIAPipeline
715716
from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
716717
from .qwenimage import (
718+
QwenImageControlNetInpaintPipeline,
717719
QwenImageControlNetPipeline,
718720
QwenImageEditInpaintPipeline,
719721
QwenImageEditPipeline,

src/diffusers/pipelines/qwenimage/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
_import_structure["modeling_qwenimage"] = ["ReduxImageEncoder"]
2626
_import_structure["pipeline_qwenimage"] = ["QwenImagePipeline"]
2727
_import_structure["pipeline_qwenimage_controlnet"] = ["QwenImageControlNetPipeline"]
28+
_import_structure["pipeline_qwenimage_controlnet_inpaint"] = ["QwenImageControlNetInpaintPipeline"]
2829
_import_structure["pipeline_qwenimage_edit"] = ["QwenImageEditPipeline"]
2930
_import_structure["pipeline_qwenimage_edit_inpaint"] = ["QwenImageEditInpaintPipeline"]
3031
_import_structure["pipeline_qwenimage_img2img"] = ["QwenImageImg2ImgPipeline"]
@@ -39,6 +40,7 @@
3940
else:
4041
from .pipeline_qwenimage import QwenImagePipeline
4142
from .pipeline_qwenimage_controlnet import QwenImageControlNetPipeline
43+
from .pipeline_qwenimage_controlnet_inpaint import QwenImageControlNetInpaintPipeline
4244
from .pipeline_qwenimage_edit import QwenImageEditPipeline
4345
from .pipeline_qwenimage_edit_inpaint import QwenImageEditInpaintPipeline
4446
from .pipeline_qwenimage_img2img import QwenImageImg2ImgPipeline

0 commit comments

Comments
 (0)