add sd_prevent_unmasked_area

2023-11-15 09:10:13 +08:00
parent f13bfb9085
commit 3e7a553273
4 changed files with 15 additions and 6 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -151,6 +151,7 @@ class SD(DiffusionInpaintModel):
            height=img_h,
            width=img_w,
            generator=torch.manual_seed(config.sd_seed),
+            callback_steps=1,
        ).images[0]

        output = (output * 255).round().astype("uint8")