"git@developer.sourcefind.cn:OpenDAS/fairseq.git" did not exist on "d4c9136ce0d6101f52307098595addfcc6a53db5"
Unverified Commit 67ffa703 authored by Nguyễn Trọng Tuấn's avatar Nguyễn Trọng Tuấn Committed by GitHub
Browse files

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: default avatar“Trgtuan10” <“tuannguyentrong.402@gmail.com”>
Co-authored-by: default avatarTuanNT-ZenAI <tuannt.zenai@gmail.com>
Co-authored-by: default avataryiyixuxu <yixu310@gmail.com>
parent 827fad66
...@@ -120,6 +120,12 @@ The `guidance_scale` parameter in the pipeline is there to support future guidan ...@@ -120,6 +120,12 @@ The `guidance_scale` parameter in the pipeline is there to support future guidan
- all - all
- __call__ - __call__
## QwenImageEditInpaintPipeline
[[autodoc]] QwenImageEditInpaintPipeline
- all
- __call__
## QwenImaggeControlNetPipeline ## QwenImaggeControlNetPipeline
- all - all
- __call__ - __call__
......
...@@ -494,6 +494,7 @@ else: ...@@ -494,6 +494,7 @@ else:
"PixArtSigmaPAGPipeline", "PixArtSigmaPAGPipeline",
"PixArtSigmaPipeline", "PixArtSigmaPipeline",
"QwenImageControlNetPipeline", "QwenImageControlNetPipeline",
"QwenImageEditInpaintPipeline",
"QwenImageEditPipeline", "QwenImageEditPipeline",
"QwenImageImg2ImgPipeline", "QwenImageImg2ImgPipeline",
"QwenImageInpaintPipeline", "QwenImageInpaintPipeline",
...@@ -1134,6 +1135,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: ...@@ -1134,6 +1135,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
PixArtSigmaPAGPipeline, PixArtSigmaPAGPipeline,
PixArtSigmaPipeline, PixArtSigmaPipeline,
QwenImageControlNetPipeline, QwenImageControlNetPipeline,
QwenImageEditInpaintPipeline,
QwenImageEditPipeline, QwenImageEditPipeline,
QwenImageImg2ImgPipeline, QwenImageImg2ImgPipeline,
QwenImageInpaintPipeline, QwenImageInpaintPipeline,
......
...@@ -393,6 +393,7 @@ else: ...@@ -393,6 +393,7 @@ else:
"QwenImageImg2ImgPipeline", "QwenImageImg2ImgPipeline",
"QwenImageInpaintPipeline", "QwenImageInpaintPipeline",
"QwenImageEditPipeline", "QwenImageEditPipeline",
"QwenImageEditInpaintPipeline",
"QwenImageControlNetPipeline", "QwenImageControlNetPipeline",
] ]
try: try:
...@@ -714,6 +715,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: ...@@ -714,6 +715,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
from .qwenimage import ( from .qwenimage import (
QwenImageControlNetPipeline, QwenImageControlNetPipeline,
QwenImageEditInpaintPipeline,
QwenImageEditPipeline, QwenImageEditPipeline,
QwenImageImg2ImgPipeline, QwenImageImg2ImgPipeline,
QwenImageInpaintPipeline, QwenImageInpaintPipeline,
......
...@@ -26,6 +26,7 @@ else: ...@@ -26,6 +26,7 @@ else:
_import_structure["pipeline_qwenimage"] = ["QwenImagePipeline"] _import_structure["pipeline_qwenimage"] = ["QwenImagePipeline"]
_import_structure["pipeline_qwenimage_controlnet"] = ["QwenImageControlNetPipeline"] _import_structure["pipeline_qwenimage_controlnet"] = ["QwenImageControlNetPipeline"]
_import_structure["pipeline_qwenimage_edit"] = ["QwenImageEditPipeline"] _import_structure["pipeline_qwenimage_edit"] = ["QwenImageEditPipeline"]
_import_structure["pipeline_qwenimage_edit_inpaint"] = ["QwenImageEditInpaintPipeline"]
_import_structure["pipeline_qwenimage_img2img"] = ["QwenImageImg2ImgPipeline"] _import_structure["pipeline_qwenimage_img2img"] = ["QwenImageImg2ImgPipeline"]
_import_structure["pipeline_qwenimage_inpaint"] = ["QwenImageInpaintPipeline"] _import_structure["pipeline_qwenimage_inpaint"] = ["QwenImageInpaintPipeline"]
...@@ -39,6 +40,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: ...@@ -39,6 +40,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
from .pipeline_qwenimage import QwenImagePipeline from .pipeline_qwenimage import QwenImagePipeline
from .pipeline_qwenimage_controlnet import QwenImageControlNetPipeline from .pipeline_qwenimage_controlnet import QwenImageControlNetPipeline
from .pipeline_qwenimage_edit import QwenImageEditPipeline from .pipeline_qwenimage_edit import QwenImageEditPipeline
from .pipeline_qwenimage_edit_inpaint import QwenImageEditInpaintPipeline
from .pipeline_qwenimage_img2img import QwenImageImg2ImgPipeline from .pipeline_qwenimage_img2img import QwenImageImg2ImgPipeline
from .pipeline_qwenimage_inpaint import QwenImageInpaintPipeline from .pipeline_qwenimage_inpaint import QwenImageInpaintPipeline
else: else:
......
...@@ -1772,6 +1772,21 @@ class QwenImageControlNetPipeline(metaclass=DummyObject): ...@@ -1772,6 +1772,21 @@ class QwenImageControlNetPipeline(metaclass=DummyObject):
requires_backends(cls, ["torch", "transformers"]) requires_backends(cls, ["torch", "transformers"])
class QwenImageEditInpaintPipeline(metaclass=DummyObject):
_backends = ["torch", "transformers"]
def __init__(self, *args, **kwargs):
requires_backends(self, ["torch", "transformers"])
@classmethod
def from_config(cls, *args, **kwargs):
requires_backends(cls, ["torch", "transformers"])
@classmethod
def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch", "transformers"])
class QwenImageEditPipeline(metaclass=DummyObject): class QwenImageEditPipeline(metaclass=DummyObject):
_backends = ["torch", "transformers"] _backends = ["torch", "transformers"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment