lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-25 01:03:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix inpaint formulation

Browse Source
This commit is contained in:
lllyasviel
2024-02-05 13:44:31 -08:00
parent f5b3fcc6cf
commit 8de4896bd6
1 changed files with 4 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
modules/sd_samplers_cfg_denoiser.py
View File

@@ -169,21 +169,9 @@ class CFGDenoiser(torch.nn.Module):
cond_composition, cond = prompt_parser.reconstruct_multicond_batch(cond, self.step)
uncond = prompt_parser.reconstruct_cond_batch(uncond, self.step)
# If we use masks, blending between the denoised and original latent images occurs here.
def apply_blend(current_latent):
blended_latent = current_latent * self.nmask + self.init_latent * self.mask
if self.p.scripts is not None:
from modules import scripts
mba = scripts.MaskBlendArgs(current_latent, self.nmask, self.init_latent, self.mask, blended_latent, denoiser=self, sigma=sigma)
self.p.scripts.on_mask_blend(self.p, mba)
blended_latent = mba.blended_latent
return blended_latent
# # Blend in the original latents (before, wrong method)
# if self.mask is not None:
# x = apply_blend(x)
if self.mask is not None:
noisy_initial_latent = self.init_latent + sigma * torch.randn_like(self.init_latent).to(self.init_latent)
x = x * self.nmask + noisy_initial_latent * self.mask
denoiser_params = CFGDenoiserParams(x, image_cond, sigma, state.sampling_step, state.sampling_steps, cond, uncond, self)
cfg_denoiser_callback(denoiser_params)
@@ -191,9 +179,8 @@ class CFGDenoiser(torch.nn.Module):
denoised = forge_sampler.forge_sample(self, denoiser_params=denoiser_params,
cond_scale=cond_scale, cond_composition=cond_composition)
# Blend in the original latents (after, correct method)
if self.mask is not None:
denoised = apply_blend(denoised)
denoised = denoised * self.nmask + self.init_latent * self.mask
preview = self.sampler.last_latent = denoised
sd_samplers_common.store_latent(preview)