Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Sign up
Appearance settings

Commit 561ab54

Browse files
haofanwanggithub-actions[bot]yiyixuxu
authored
Support ControlNet for Qwen-Image (#12215)
* support qwen-image-cn-union --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: YiYi Xu <yixu310@gmail.com>
1 parent b60faf4 commit 561ab54

File tree

12 files changed

+1382
-1
lines changed

12 files changed

+1382
-1
lines changed

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

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

123+
## QwenImaggeControlNetPipeline
124+
- all
125+
- __call__
126+
123127
## QwenImagePipelineOutput
124128

125129
[[autodoc]] pipelines.qwenimage.pipeline_output.QwenImagePipelineOutput

‎src/diffusers/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,8 @@
218218
"OmniGenTransformer2DModel",
219219
"PixArtTransformer2DModel",
220220
"PriorTransformer",
221+
"QwenImageControlNetModel",
222+
"QwenImageMultiControlNetModel",
221223
"QwenImageTransformer2DModel",
222224
"SanaControlNetModel",
223225
"SanaTransformer2DModel",
@@ -491,6 +493,7 @@
491493
"PixArtAlphaPipeline",
492494
"PixArtSigmaPAGPipeline",
493495
"PixArtSigmaPipeline",
496+
"QwenImageControlNetPipeline",
494497
"QwenImageEditPipeline",
495498
"QwenImageImg2ImgPipeline",
496499
"QwenImageInpaintPipeline",
@@ -885,6 +888,8 @@
885888
OmniGenTransformer2DModel,
886889
PixArtTransformer2DModel,
887890
PriorTransformer,
891+
QwenImageControlNetModel,
892+
QwenImageMultiControlNetModel,
888893
QwenImageTransformer2DModel,
889894
SanaControlNetModel,
890895
SanaTransformer2DModel,
@@ -1128,6 +1133,7 @@
11281133
PixArtAlphaPipeline,
11291134
PixArtSigmaPAGPipeline,
11301135
PixArtSigmaPipeline,
1136+
QwenImageControlNetPipeline,
11311137
QwenImageEditPipeline,
11321138
QwenImageImg2ImgPipeline,
11331139
QwenImageInpaintPipeline,

‎src/diffusers/models/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,10 @@
5252
"HunyuanDiT2DControlNetModel",
5353
"HunyuanDiT2DMultiControlNetModel",
5454
]
55+
_import_structure["controlnets.controlnet_qwenimage"] = [
56+
"QwenImageControlNetModel",
57+
"QwenImageMultiControlNetModel",
58+
]
5559
_import_structure["controlnets.controlnet_sana"] = ["SanaControlNetModel"]
5660
_import_structure["controlnets.controlnet_sd3"] = ["SD3ControlNetModel", "SD3MultiControlNetModel"]
5761
_import_structure["controlnets.controlnet_sparsectrl"] = ["SparseControlNetModel"]
@@ -148,6 +152,8 @@
148152
HunyuanDiT2DMultiControlNetModel,
149153
MultiControlNetModel,
150154
MultiControlNetUnionModel,
155+
QwenImageControlNetModel,
156+
QwenImageMultiControlNetModel,
151157
SanaControlNetModel,
152158
SD3ControlNetModel,
153159
SD3MultiControlNetModel,

‎src/diffusers/models/controlnets/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
HunyuanDiT2DControlNetModel,
1010
HunyuanDiT2DMultiControlNetModel,
1111
)
12+
from .controlnet_qwenimage import QwenImageControlNetModel, QwenImageMultiControlNetModel
1213
from .controlnet_sana import SanaControlNetModel
1314
from .controlnet_sd3 import SD3ControlNetModel, SD3ControlNetOutput, SD3MultiControlNetModel
1415
from .controlnet_sparsectrl import (

0 commit comments

Comments
(0)

AltStyle によって変換されたページ (->オリジナル) /