Skip to content

Commit 67ffa70

Browse files
Trgtuan10“Trgtuan10”TuanNT-ZenAIyiyixuxu
authored
Add Qwen-Image-Edit Inpainting pipeline (#12225)
* add qwenimage-edit inpaint feature * stay up to date with main branch * fix style * fix docs * copies * fix * again * copies --------- Co-authored-by: “Trgtuan10” <“tuannguyentrong.402@gmail.com”> Co-authored-by: TuanNT-ZenAI <tuannt.zenai@gmail.com> Co-authored-by: yiyixuxu <yixu310@gmail.com>
1 parent 827fad6 commit 67ffa70

File tree

6 files changed

+1133
-0
lines changed

6 files changed

+1133
-0
lines changed

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,12 @@ The `guidance_scale` parameter in the pipeline is there to support future guidan
120120
- all
121121
- __call__
122122

123+
## QwenImageEditInpaintPipeline
124+
125+
[[autodoc]] QwenImageEditInpaintPipeline
126+
- all
127+
- __call__
128+
123129
## QwenImaggeControlNetPipeline
124130
- all
125131
- __call__

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -494,6 +494,7 @@
494494
"PixArtSigmaPAGPipeline",
495495
"PixArtSigmaPipeline",
496496
"QwenImageControlNetPipeline",
497+
"QwenImageEditInpaintPipeline",
497498
"QwenImageEditPipeline",
498499
"QwenImageImg2ImgPipeline",
499500
"QwenImageInpaintPipeline",
@@ -1134,6 +1135,7 @@
11341135
PixArtSigmaPAGPipeline,
11351136
PixArtSigmaPipeline,
11361137
QwenImageControlNetPipeline,
1138+
QwenImageEditInpaintPipeline,
11371139
QwenImageEditPipeline,
11381140
QwenImageImg2ImgPipeline,
11391141
QwenImageInpaintPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,6 +393,7 @@
393393
"QwenImageImg2ImgPipeline",
394394
"QwenImageInpaintPipeline",
395395
"QwenImageEditPipeline",
396+
"QwenImageEditInpaintPipeline",
396397
"QwenImageControlNetPipeline",
397398
]
398399
try:
@@ -714,6 +715,7 @@
714715
from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
715716
from .qwenimage import (
716717
QwenImageControlNetPipeline,
718+
QwenImageEditInpaintPipeline,
717719
QwenImageEditPipeline,
718720
QwenImageImg2ImgPipeline,
719721
QwenImageInpaintPipeline,

src/diffusers/pipelines/qwenimage/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
_import_structure["pipeline_qwenimage"] = ["QwenImagePipeline"]
2727
_import_structure["pipeline_qwenimage_controlnet"] = ["QwenImageControlNetPipeline"]
2828
_import_structure["pipeline_qwenimage_edit"] = ["QwenImageEditPipeline"]
29+
_import_structure["pipeline_qwenimage_edit_inpaint"] = ["QwenImageEditInpaintPipeline"]
2930
_import_structure["pipeline_qwenimage_img2img"] = ["QwenImageImg2ImgPipeline"]
3031
_import_structure["pipeline_qwenimage_inpaint"] = ["QwenImageInpaintPipeline"]
3132

@@ -39,6 +40,7 @@
3940
from .pipeline_qwenimage import QwenImagePipeline
4041
from .pipeline_qwenimage_controlnet import QwenImageControlNetPipeline
4142
from .pipeline_qwenimage_edit import QwenImageEditPipeline
43+
from .pipeline_qwenimage_edit_inpaint import QwenImageEditInpaintPipeline
4244
from .pipeline_qwenimage_img2img import QwenImageImg2ImgPipeline
4345
from .pipeline_qwenimage_inpaint import QwenImageInpaintPipeline
4446
else:

0 commit comments

Comments
 (0)