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

Add gradient blending to tile seams in MultiDiffusion.

Browse Source
This commit is contained in:
Ryan Dick
2024-07-19 10:03:01 -04:00
committed by Kent Keirsey
parent 97a7f51721
commit e16faa6370
2 changed files with 43 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/invocations/tiled_multi_diffusion_denoise_latents.py
View File

@ -175,6 +175,10 @@ class TiledMultiDiffusionDenoiseLatents(BaseInvocation):
_, _, latent_height, latent_width = latents.shape
# Calculate the tile locations to cover the latent-space image.
# TODO(ryand): In the future, we may want to revisit the tile overlap strategy. Things to consider:
# - How much overlap 'context' to provide for each denoising step.
# - How much overlap to use during merging/blending.
# - Should we 'jitter' the tile locations in each step so that the seams are in different places?
tiles = calc_tiles_min_overlap(
image_height=latent_height,
image_width=latent_width,