diff options
| -rw-r--r-- | modules/sd_samplers.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 1b3dc302..af6811a1 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -255,6 +255,7 @@ class KDiffusionSampler: self.model_wrap_cfg.nmask = p.nmask
self.model_wrap_cfg.init_latent = p.init_latent
self.model_wrap.step = 0
+ self.sampler_noise_index = 0
if hasattr(k_diffusion.sampling, 'trange'):
k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs)
@@ -269,6 +270,7 @@ class KDiffusionSampler: x = x * sigmas[0]
self.model_wrap_cfg.step = 0
+ self.sampler_noise_index = 0
if hasattr(k_diffusion.sampling, 'trange'):
k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs)
|