From da3898ef4f3e0ed23dc1d61019f8b66fc41ae811 Mon Sep 17 00:00:00 2001 From: hlky Date: Fri, 11 Oct 2024 09:24:11 +0100 Subject: [PATCH] FluxMultiControlNetModel --- src/diffusers/pipelines/flux/pipeline_flux_controlnet.py | 2 ++ .../pipelines/flux/pipeline_flux_controlnet_image_to_image.py | 2 ++ .../pipelines/flux/pipeline_flux_controlnet_inpainting.py | 2 ++ 3 files changed, 6 insertions(+) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py index a301f6742c05..8770c231809f 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet.py @@ -202,6 +202,8 @@ def __init__( ], ): super().__init__() + if isinstance(controlnet, (list, tuple)): + controlnet = FluxMultiControlNetModel(controlnet) self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py index b5ff2236c4d0..a7f7c66a2cad 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py @@ -214,6 +214,8 @@ def __init__( ], ): super().__init__() + if isinstance(controlnet, (list, tuple)): + controlnet = FluxMultiControlNetModel(controlnet) self.register_modules( vae=vae, diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py index 1c1c25302100..50d2fcaa7fa5 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py @@ -216,6 +216,8 @@ def __init__( ], ): super().__init__() + if isinstance(controlnet, (list, tuple)): + controlnet = FluxMultiControlNetModel(controlnet) self.register_modules( scheduler=scheduler,