Skip to content

Commit 0f8fb75

Browse files
authored
FluxMultiControlNetModel (#9647)
1 parent 3033f08 commit 0f8fb75

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

src/diffusers/pipelines/flux/pipeline_flux_controlnet.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,8 @@ def __init__(
202202
],
203203
):
204204
super().__init__()
205+
if isinstance(controlnet, (list, tuple)):
206+
controlnet = FluxMultiControlNetModel(controlnet)
205207

206208
self.register_modules(
207209
vae=vae,

src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,8 @@ def __init__(
214214
],
215215
):
216216
super().__init__()
217+
if isinstance(controlnet, (list, tuple)):
218+
controlnet = FluxMultiControlNetModel(controlnet)
217219

218220
self.register_modules(
219221
vae=vae,

src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -216,6 +216,8 @@ def __init__(
216216
],
217217
):
218218
super().__init__()
219+
if isinstance(controlnet, (list, tuple)):
220+
controlnet = FluxMultiControlNetModel(controlnet)
219221

220222
self.register_modules(
221223
scheduler=scheduler,

0 commit comments

Comments
 (0)