lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-26 01:09:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix inpaint batch dim align #94

Browse Source
This commit is contained in:
lllyasviel
2024-02-06 22:57:53 -08:00
parent 65f9c7d442
commit 4ea4a92fe9
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_cfg_denoiser.py
View File

@@ -173,7 +173,7 @@ class CFGDenoiser(torch.nn.Module):
uncond = prompt_parser.reconstruct_cond_batch(uncond, self.step)
if self.mask is not None:
noisy_initial_latent = self.init_latent + sigma * torch.randn_like(self.init_latent).to(self.init_latent)
noisy_initial_latent = self.init_latent + sigma[:, None, None, None] * 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)