"docs/vscode:/vscode.git/clone" did not exist on "ca35aa79e2989cbdc51ce2781bf32dcf267af322"
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
],
):
super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = FluxMultiControlNetModel(controlnet)
self.register_modules(
vae=vae,
......
......@@ -214,6 +214,8 @@ class FluxControlNetImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From
],
):
super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = FluxMultiControlNetModel(controlnet)
self.register_modules(
vae=vae,
......
......@@ -216,6 +216,8 @@ class FluxControlNetInpaintPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From
],
):
super().__init__()
if isinstance(controlnet, (list, tuple)):
controlnet = FluxMultiControlNetModel(controlnet)
self.register_modules(
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