aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2024-01-01 11:11:14 +0300
committerAUTOMATIC1111 <16777216c@gmail.com>2024-01-01 11:11:14 +0300
commit5692bf1517c3409ad46262c56e65f256389825b1 (patch)
treeddd0dfcbc0713e317fba79b1fd1cd8e6f80e6b24
parente55fec9d9a8abf01d37ecb80f3fb1e521afe0d0b (diff)
add missing field for DDIM sampler that was breaking img2img
-rw-r--r--modules/sd_samplers_timesteps.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/sd_samplers_timesteps.py b/modules/sd_samplers_timesteps.py
index b17a8f93..f8afa8bd 100644
--- a/modules/sd_samplers_timesteps.py
+++ b/modules/sd_samplers_timesteps.py
@@ -80,6 +80,7 @@ class CompVisSampler(sd_samplers_common.Sampler):
self.eta_default = 0.0
self.model_wrap_cfg = CFGDenoiserTimesteps(self)
+ self.model_wrap = self.model_wrap_cfg.inner_model
def get_timesteps(self, p, steps):
discard_next_to_last_sigma = self.config is not None and self.config.options.get('discard_next_to_last_sigma', False)