Skip to content

Commit df267ee

Browse files
authored
feat: Add QwenImageEditPlus to support future feature upgrades (#12357)
* feat: add support of qwenimageeditplus * add copies statement * fix copies statement * remove vl_processor reference
1 parent edd614e commit df267ee

File tree

6 files changed

+904
-1
lines changed

6 files changed

+904
-1
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -515,6 +515,7 @@
515515
"QwenImageControlNetPipeline",
516516
"QwenImageEditInpaintPipeline",
517517
"QwenImageEditPipeline",
518+
"QwenImageEditPlusPipeline",
518519
"QwenImageImg2ImgPipeline",
519520
"QwenImageInpaintPipeline",
520521
"QwenImagePipeline",
@@ -1170,6 +1171,7 @@
11701171
QwenImageControlNetPipeline,
11711172
QwenImageEditInpaintPipeline,
11721173
QwenImageEditPipeline,
1174+
QwenImageEditPlusPipeline,
11731175
QwenImageImg2ImgPipeline,
11741176
QwenImageInpaintPipeline,
11751177
QwenImagePipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -394,6 +394,7 @@
394394
"QwenImageImg2ImgPipeline",
395395
"QwenImageInpaintPipeline",
396396
"QwenImageEditPipeline",
397+
"QwenImageEditPlusPipeline",
397398
"QwenImageEditInpaintPipeline",
398399
"QwenImageControlNetInpaintPipeline",
399400
"QwenImageControlNetPipeline",
@@ -721,6 +722,7 @@
721722
QwenImageControlNetPipeline,
722723
QwenImageEditInpaintPipeline,
723724
QwenImageEditPipeline,
725+
QwenImageEditPlusPipeline,
724726
QwenImageImg2ImgPipeline,
725727
QwenImageInpaintPipeline,
726728
QwenImagePipeline,

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_controlnet"] = ["QwenImageControlNetPipeline"]
2828
_import_structure["pipeline_qwenimage_controlnet_inpaint"] = ["QwenImageControlNetInpaintPipeline"]
2929
_import_structure["pipeline_qwenimage_edit"] = ["QwenImageEditPipeline"]
30+
_import_structure["pipeline_qwenimage_edit_plus"] = ["QwenImageEditPlusPipeline"]
3031
_import_structure["pipeline_qwenimage_edit_inpaint"] = ["QwenImageEditInpaintPipeline"]
3132
_import_structure["pipeline_qwenimage_img2img"] = ["QwenImageImg2ImgPipeline"]
3233
_import_structure["pipeline_qwenimage_inpaint"] = ["QwenImageInpaintPipeline"]
@@ -43,6 +44,7 @@
4344
from .pipeline_qwenimage_controlnet_inpaint import QwenImageControlNetInpaintPipeline
4445
from .pipeline_qwenimage_edit import QwenImageEditPipeline
4546
from .pipeline_qwenimage_edit_inpaint import QwenImageEditInpaintPipeline
47+
from .pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
4648
from .pipeline_qwenimage_img2img import QwenImageImg2ImgPipeline
4749
from .pipeline_qwenimage_inpaint import QwenImageInpaintPipeline
4850
else:

src/diffusers/pipelines/qwenimage/pipeline_qwenimage_edit.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,6 @@ def __init__(
208208
# QwenImage latents are turned into 2x2 patches and packed. This means the latent width and height has to be divisible
209209
# by the patch size. So the vae scale factor is multiplied by the patch size to account for this
210210
self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor * 2)
211-
self.vl_processor = processor
212211
self.tokenizer_max_length = 1024
213212

214213
self.prompt_template_encode = "<|im_start|>system\nDescribe the key features of the input image (color, shape, size, texture, objects, background), then explain how the user's text instruction should alter or modify the image. Generate a new image that meets the user's requirements while maintaining consistency with the original input where appropriate.<|im_end|>\n<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>{}<|im_end|>\n<|im_start|>assistant\n"

0 commit comments

Comments
 (0)