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

fix flux img2img and highres.fix math

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

8
modules/sd_samplers_kdiffusion.py
View File

@@ -139,7 +139,8 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
sigma_sched = sigmas[steps - t_enc - 1:]
x = x.to(noise)
xi = x + noise * sigma_sched[0]
xi = self.model_wrap.predictor.noise_scaling(sigma_sched[0], noise, x, max_denoise=False)
if opts.img2img_extra_noise > 0:
p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise
@@ -198,9 +199,8 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
if opts.sgm_noise_multiplier:
p.extra_generation_params["SGM noise multiplier"] = True
x = x * torch.sqrt(1.0 + sigmas[0] ** 2.0)
else:
x = x * sigmas[0]
x = self.model_wrap.predictor.noise_scaling(sigmas[0], x, torch.zeros_like(x), max_denoise=opts.sgm_noise_multiplier)
extra_params_kwargs = self.initialize(p)
parameters = inspect.signature(self.func).parameters