Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Switch from dilate to erode to fix inpaint edges. Default model to 1.4 instead of 1.5.

Browse Source
This commit is contained in:
Kyle Schouviller 2022-10-23 14:01:06 -07:00
parent 3b01e6e423
commit 1264cc2d36
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
configs/models.yaml
View File

@ -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

2
ldm/invoke/generator/inpaint.py
View File

@ -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: