"src/vscode:/vscode.git/clone" did not exist on "046dc4307507b02b8514c63c7664b723da84f2ac"
Unverified Commit 957e5cab authored by hlky's avatar hlky Committed by GitHub
Browse files

Convert list/tuple of `HunyuanDiT2DControlNetModel` to `HunyuanDiT2DMultiControlNetModel` (#9651)



Convert list/tuple of HunyuanDiT2DControlNetModel to HunyuanDiT2DMultiControlNetModel
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent 3e4c5707
......@@ -225,6 +225,8 @@ class HunyuanDiTControlNetPipeline(DiffusionPipeline):
requires_safety_checker: bool = True,
):
super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = HunyuanDiT2DMultiControlNetModel(controlnet)
self.register_modules(
vae=vae,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment