switch controlnet in webui

2023-05-13 13:45:27 +08:00
parent 0363472adc
commit 3eef8f4dae
16 changed files with 161 additions and 46 deletions
--- a/lama_cleaner/model/controlnet.py
+++ b/lama_cleaner/model/controlnet.py
@@ -203,6 +203,7 @@ class ControlNet(DiffusionInpaintModel):
                negative_prompt=config.negative_prompt,
                generator=torch.manual_seed(config.sd_seed),
                output_type="np.array",
+                callback=self.callback
            ).images[0]
        else:
            if "canny" in self.sd_controlnet_method: