pass pipe_components when turn control on/off

2024-01-09 22:58:21 +08:00
parent db4a6f4547
commit 05a15b2e1f
6 changed files with 26 additions and 9 deletions
--- a/iopaint/model/controlnet.py
+++ b/iopaint/model/controlnet.py
@@ -47,7 +47,7 @@ class ControlNet(DiffusionInpaintModel):
        self.model_info = model_info
        self.controlnet_method = controlnet_method

-        model_kwargs = {}
+        model_kwargs = {**kwargs.get("pipe_components", {})}
        if kwargs["disable_nsfw"] or kwargs.get("cpu_offload", False):
            logger.info("Disable Stable Diffusion Model NSFW checker")
            model_kwargs.update(