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/sd.py
+++ b/iopaint/model/sd.py
@@ -19,7 +19,7 @@ class SD(DiffusionInpaintModel):

        use_gpu, torch_dtype = get_torch_dtype(device, kwargs.get("no_half", False))

-        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(