add sd_prevent_unmasked_area

lama_cleaner/model/base.py

@@ -67,8 +67,9 @@ class InpaintModel:
 
         result, image, mask = self.forward_post_process(result, image, mask, config)
 
-        mask = mask[:, :, np.newaxis]
-        result = result * (mask / 255) + image[:, :, ::-1] * (1 - (mask / 255))
+        if config.sd_prevent_unmasked_area:
+            mask = mask[:, :, np.newaxis]
+            result = result * (mask / 255) + image[:, :, ::-1] * (1 - (mask / 255))
         return result
 
     def forward_post_process(self, result, image, mask, config):

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")