Skip to content

Commit 3097624

Browse files
author
“Trgtuan10”
committed
add qwenimage-edit inpaint feature
1 parent cc48b93 commit 3097624

File tree

5 files changed

+1084
-0
lines changed

5 files changed

+1084
-0
lines changed

src/diffusers/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -489,6 +489,7 @@
489489
"PixArtAlphaPipeline",
490490
"PixArtSigmaPAGPipeline",
491491
"PixArtSigmaPipeline",
492+
"QwenImageEditInpaintPipeline",
492493
"QwenImageEditPipeline",
493494
"QwenImageImg2ImgPipeline",
494495
"QwenImageInpaintPipeline",

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -392,6 +392,7 @@
392392
"QwenImageImg2ImgPipeline",
393393
"QwenImageInpaintPipeline",
394394
"QwenImageEditPipeline",
395+
"QwenImageEditInpaintPipeline",
395396
]
396397
try:
397398
if not is_onnx_available():
@@ -710,6 +711,7 @@
710711
from .pia import PIAPipeline
711712
from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
712713
from .qwenimage import (
714+
QwenImageEditInpaintPipeline,
713715
QwenImageEditPipeline,
714716
QwenImageImg2ImgPipeline,
715717
QwenImageInpaintPipeline,

src/diffusers/pipelines/qwenimage/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
_import_structure["pipeline_qwenimage_edit"] = ["QwenImageEditPipeline"]
2828
_import_structure["pipeline_qwenimage_img2img"] = ["QwenImageImg2ImgPipeline"]
2929
_import_structure["pipeline_qwenimage_inpaint"] = ["QwenImageInpaintPipeline"]
30+
_import_structure["pipeline_qwenimage_edit_inpaint"] = ["QwenImageEditInpaintPipeline"]
3031

3132
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
3233
try:
@@ -39,6 +40,7 @@
3940
from .pipeline_qwenimage_edit import QwenImageEditPipeline
4041
from .pipeline_qwenimage_img2img import QwenImageImg2ImgPipeline
4142
from .pipeline_qwenimage_inpaint import QwenImageInpaintPipeline
43+
from .pipeline_qwenimage_edit_inpaint import QwenImageEditInpaintPipeline
4244
else:
4345
import sys
4446

0 commit comments

Comments
 (0)