lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 18:51:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

make flux inpaint and soft inpaint works

Browse Source
This commit is contained in:
layerdiffusion
2024-08-11 18:18:16 -07:00
parent 36250c78f5
commit cd8a181438
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/sd_samplers_cfg_denoiser.py
View File

@@ -176,7 +176,8 @@ class CFGDenoiser(torch.nn.Module):
uncond = prompt_parser.reconstruct_cond_batch(uncond, self.step) if uncond is not None else None uncond = prompt_parser.reconstruct_cond_batch(uncond, self.step) if uncond is not None else None
if self.mask is not None: if self.mask is not None:
noisy_initial_latent = self.init_latent + sigma[:, None, None, None] * torch.randn_like(self.init_latent).to(self.init_latent) predictor = self.inner_model.inner_model.forge_objects.unet.model.predictor
noisy_initial_latent = predictor.noise_scaling(sigma[:, None, None, None], torch.randn_like(self.init_latent).to(self.init_latent), self.init_latent, max_denoise=False)
x = x * self.nmask + noisy_initial_latent * self.mask 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) denoiser_params = CFGDenoiserParams(x, image_cond, sigma, state.sampling_step, state.sampling_steps, cond, uncond, self)