Skip to content

Commit b0ed112

Browse files
committed
Rename extra noise variables to be clearer
1 parent 369655e commit b0ed112

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

modules/sd_samplers_kdiffusion.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -145,9 +145,9 @@ def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning,
145145

146146
if opts.img2img_extra_noise > 0:
147147
p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise
148-
start_code = x + (noise * opts.img2img_extra_noise)
148+
start_noise = x + (noise * opts.img2img_extra_noise)
149149
noise = noise * sigma_sched[0]
150-
xi = start_code + noise
150+
xi = start_noise + noise
151151

152152
extra_params_kwargs = self.initialize(p)
153153
parameters = inspect.signature(self.func).parameters

modules/sd_samplers_timesteps.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -103,9 +103,9 @@ def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning,
103103

104104
if opts.img2img_extra_noise > 0:
105105
p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise
106-
start_code = x + (noise * opts.img2img_extra_noise)
106+
start_noise = x + (noise * opts.img2img_extra_noise)
107107
noise = noise * sqrt_one_minus_alpha_cumprod
108-
xi = start_code * sqrt_alpha_cumprod + noise
108+
xi = start_noise * sqrt_alpha_cumprod + noise
109109

110110
extra_params_kwargs = self.initialize(p)
111111
parameters = inspect.signature(self.func).parameters

0 commit comments

Comments
 (0)