Unverified Commit 0f8fb75c authored by hlky's avatar hlky Committed by GitHub
Browse files

FluxMultiControlNetModel (#9647)

parent 3033f082
...@@ -202,6 +202,8 @@ class FluxControlNetPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleF ...@@ -202,6 +202,8 @@ class FluxControlNetPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleF
], ],
): ):
super().__init__() super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = FluxMultiControlNetModel(controlnet)
self.register_modules( self.register_modules(
vae=vae, vae=vae,
......
...@@ -214,6 +214,8 @@ class FluxControlNetImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From ...@@ -214,6 +214,8 @@ class FluxControlNetImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From
], ],
): ):
super().__init__() super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = FluxMultiControlNetModel(controlnet)
self.register_modules( self.register_modules(
vae=vae, vae=vae,
......
...@@ -216,6 +216,8 @@ class FluxControlNetInpaintPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From ...@@ -216,6 +216,8 @@ class FluxControlNetInpaintPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From
], ],
): ):
super().__init__() super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = FluxMultiControlNetModel(controlnet)
self.register_modules( self.register_modules(
scheduler=scheduler, scheduler=scheduler,
......
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