Skip to content

Commit 957e5ca

Browse files
hlkyyiyixuxu
andauthored
Convert list/tuple of HunyuanDiT2DControlNetModel to HunyuanDiT2DMultiControlNetModel (huggingface#9651)
Convert list/tuple of HunyuanDiT2DControlNetModel to HunyuanDiT2DMultiControlNetModel Co-authored-by: YiYi Xu <[email protected]>
1 parent 3e4c570 commit 957e5ca

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/pipelines/controlnet_hunyuandit/pipeline_hunyuandit_controlnet.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,8 @@ def __init__(
225225
requires_safety_checker: bool = True,
226226
):
227227
super().__init__()
228+
if isinstance(controlnet, (list, tuple)):
229+
controlnet = HunyuanDiT2DMultiControlNetModel(controlnet)
228230

229231
self.register_modules(
230232
vae=vae,

0 commit comments

Comments
 (0)