aboutsummaryrefslogtreecommitdiff
path: root/modules
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-01-25 23:25:40 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-01-25 23:25:40 +0300
commit6cff4401824299a983c8e13424018efc347b4a2b (patch)
treefe8ed71cea13ade8a1b01f53ba2add11969a55bb /modules
parentd1d6ce29831d1b067801c3206f314258de88f683 (diff)
fix prompt editing break after first batch in img2img
Diffstat (limited to 'modules')
-rw-r--r--modules/sd_samplers.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py
index 6261d1f7..a7910b56 100644
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@@ -454,7 +454,7 @@ class KDiffusionSampler:
def initialize(self, p):
self.model_wrap_cfg.mask = p.mask if hasattr(p, 'mask') else None
self.model_wrap_cfg.nmask = p.nmask if hasattr(p, 'nmask') else None
- self.model_wrap.step = 0
+ self.model_wrap_cfg.step = 0
self.eta = p.eta or opts.eta_ancestral
k_diffusion.sampling.torch = TorchHijack(self.sampler_noises if self.sampler_noises is not None else [])