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

fix(inpainting model): threshold mask to avoid gray blurry seam

Browse Source
This commit is contained in:
Kevin Turner 2023-01-28 13:34:22 -08:00
parent 71b6ddf5fb
commit b6d37a70ca
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ldm/invoke/generator/diffusers_pipeline.py
View File

@ -556,12 +556,12 @@ class StableDiffusionGeneratorPipeline(StableDiffusionPipeline):
guidance: List[Callable] = []
if is_inpainting_model(self.unet):
# TODO: we should probably pass this in so we don't have to try/finally around setting it.
# You'd think the inpainting model wouldn't be paying attention to the area it is going to repaint
# (that's why there's a mask!) but it seems to really want that blanked out.
masked_latents = self.non_noised_latents_from_image(
init_image * (1 - mask), device=device, dtype=latents_dtype)
masked_init_image = init_image * torch.where(mask < 0.5, 1, 0)
masked_latents = self.non_noised_latents_from_image(masked_init_image, device=device, dtype=latents_dtype)
# TODO: we should probably pass this in so we don't have to try/finally around setting it.
self.invokeai_diffuser.model_forward_callback = \
AddsMaskLatents(self._unet_forward, latent_mask, masked_latents)
else: