sd add strength

2023-11-14 14:02:15 +08:00 · 2023-11-14 14:02:15 +08:00 · 53b9f3e92f
commit 53b9f3e92f
parent d061b07029
1 changed files with 1 additions and 0 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@ -146,6 +146,7 @@ class SD(DiffusionInpaintModel):
            negative_prompt=config.negative_prompt,
            mask_image=PIL.Image.fromarray(mask[:, :, -1], mode="L"),
            num_inference_steps=config.sd_steps,
+            strength=config.sd_strength,
            guidance_scale=config.sd_guidance_scale,
            output_type="np",
            callback=self.callback,