Skip to content

GitLab

"src/diffusers/pipelines/+" did not exist on "c2d76da33768b60fa0dcc6859c34067d66f00558"
Unverified Commit 3e4c5707 authored by hlky's avatar hlky Committed by GitHub
Browse files

Convert list/tuple of `SD3ControlNetModel` to `SD3MultiControlNetModel` (#9652) 



Convert list/tuple of SD3ControlNetModel to SD3MultiControlNetModel
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent 1bcd19e4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/pipelines/controlnet_sd3/pipeline_stable_diffusion_3_controlnet.py
View file @ 3e4c5707
...@@ -192,6 +192,8 @@ class StableDiffusion3ControlNetPipeline(DiffusionPipeline, SD3LoraLoaderMixin, ...@@ -192,6 +192,8 @@ class StableDiffusion3ControlNetPipeline(DiffusionPipeline, SD3LoraLoaderMixin,
], ],
): ):
super().__init__() super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = SD3MultiControlNetModel(controlnet)
self.register_modules( self.register_modules(
vae=vae, 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