diff --git a/configs/models.yaml b/configs/models.yaml index f3fde45d8f..3fea907f4c 100644 --- a/configs/models.yaml +++ b/configs/models.yaml @@ -12,6 +12,7 @@ stable-diffusion-1.4: description: Stable Diffusion inference model version 1.4 width: 512 height: 512 + default: true stable-diffusion-1.5: config: configs/stable-diffusion/v1-inference.yaml weights: models/ldm/stable-diffusion-v1/v1-5-pruned-emaonly.ckpt diff --git a/ldm/invoke/generator/inpaint.py b/ldm/invoke/generator/inpaint.py index ee67b90c46..52aaa153ee 100644 --- a/ldm/invoke/generator/inpaint.py +++ b/ldm/invoke/generator/inpaint.py @@ -111,7 +111,7 @@ class Inpaint(Img2Img): # Blur the mask out (into init image) by specified amount if mask_blur_radius > 0: nm = np.asarray(pil_init_mask, dtype=np.uint8) - nmd = cv.dilate(nm, kernel=np.ones((3,3), dtype=np.uint8), iterations=int(mask_blur_radius / 2)) + nmd = cv.erode(nm, kernel=np.ones((3,3), dtype=np.uint8), iterations=int(mask_blur_radius / 2)) pmd = Image.fromarray(nmd, mode='L') blurred_init_mask = pmd.filter(ImageFilter.BoxBlur(mask_blur_radius)) else: