From 957e5cabff0d6f0815a162d0449d42568722210b Mon Sep 17 00:00:00 2001
From: hlky <hlky@hlky.ac>
Date: Tue, 15 Oct 2024 05:09:30 +0100
Subject: [PATCH] Convert list/tuple of `HunyuanDiT2DControlNetModel` to
 `HunyuanDiT2DMultiControlNetModel` (#9651)

Convert list/tuple of HunyuanDiT2DControlNetModel to HunyuanDiT2DMultiControlNetModel

Co-authored-by: YiYi Xu <yixu310@gmail.com>
---
 .../controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py b/src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py
index 10c521e84997..2cbbea08ef16 100644
--- 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,