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

Update sd_samplers_timesteps.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 08:18:56 -08:00
parent dbdc861d62
commit 9ab5589a62
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_timesteps.py
View File

@@ -115,7 +115,7 @@ class CompVisSampler(sd_samplers_common.Sampler):
sqrt_alpha_cumprod = torch.sqrt(alphas_cumprod[timesteps[t_enc]])
sqrt_one_minus_alpha_cumprod = torch.sqrt(1 - alphas_cumprod[timesteps[t_enc]])
xi = x * sqrt_alpha_cumprod + noise * sqrt_one_minus_alpha_cumprod
xi = x.to(noise) * sqrt_alpha_cumprod + noise * sqrt_one_minus_alpha_cumprod
if opts.img2img_extra_noise > 0:
p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise