Convert list/tuple of HunyuanDiT2DControlNetModel to `HunyuanDiT2DM…

…ultiControlNetModel` (#9651) Convert list/tuple of HunyuanDiT2DControlNetModel to HunyuanDiT2DMultiControlNetModel Co-authored-by: YiYi Xu <yixu310@gmail.com>
huggingface · Oct 15, 2024 · 957e5ca · 957e5ca
1 parent 3e4c570
commit 957e5ca
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py b/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py
@@ -225,6 +225,8 @@ def __init__(
         requires_safety_checker: bool = True,
     ):
         super().__init__()
+        if isinstance(controlnet, (list, tuple)):
+            controlnet = HunyuanDiT2DMultiControlNetModel(controlnet)
 
         self.register_modules(
             vae=vae,