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

Update sd_samplers_cfg_denoiser.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 19:44:25 -08:00
parent 54f07f1d1d
commit d3cb546cc5
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sd_samplers_cfg_denoiser.py
View File

@@ -90,9 +90,11 @@ class CFGDenoiser(torch.nn.Module):
cfg_result = x - model_options["sampler_cfg_function"](args) cfg_result = x - model_options["sampler_cfg_function"](args)
else: else:
cfg_result = uncond_pred + (cond_pred - uncond_pred) * cond_scale cfg_result = uncond_pred + (cond_pred - uncond_pred) * cond_scale
# sanity_check = torch.allclose(cfg_result, denoised)
for fn in model_options.get("sampler_post_cfg_function", []): for fn in model_options.get("sampler_post_cfg_function", []):
args = {"denoised": cfg_result, "cond": torch.zeros_like(uncond), "uncond": torch.zeros_like(uncond), "model": model, args = {"denoised": cfg_result, "cond": torch.zeros_like(uncond),
"uncond": torch.zeros_like(uncond), "model": model,
"uncond_denoised": uncond_pred, "cond_denoised": cond_pred, "uncond_denoised": uncond_pred, "cond_denoised": cond_pred,
"sigma": timestep, "model_options": model_options, "input": x} "sigma": timestep, "model_options": model_options, "input": x}
cfg_result = fn(args) cfg_result = fn(args)