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

Don't try to tile fill if image doesn't have an alpha layer

Browse Source
This commit is contained in:
Kyle Schouviller 2022-10-27 17:49:26 -07:00 committed by psychedelicious
parent 320cbdd62d
commit 43de16cae4
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -43,6 +43,10 @@ class Inpaint(Img2Img):
)
def tile_fill_missing(self, im: Image.Image, tile_size: int = 16, seed: int = None) -> Image:
# Only fill if there's an alpha layer
if im.mode != 'RGBA':
return im
a = np.asarray(im, dtype=np.uint8)
tile_size = (tile_size, tile_size)