add negative prompt

2022-11-08 21:58:48 +08:00
parent b7d504cba6
commit 8c2904c9c8
13 changed files with 176 additions and 8 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -140,6 +140,7 @@ class SD(InpaintModel):

        output = self.model(
            prompt=config.prompt,
+            negative_prompt=config.negative_prompt,
            mask_image=PIL.Image.fromarray(mask[:, :, -1], mode="L"),
            strength=config.sd_strength,
            num_inference_steps=config.sd_steps,