aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--webui.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/webui.py b/webui.py
index 3c6611ed..018da2f3 100644
--- a/webui.py
+++ b/webui.py
@@ -852,7 +852,7 @@ def p_sample_ddim_hook(sampler_wrapper, x_dec, cond, ts, *args, **kwargs):
class VanillaStableDiffusionSampler:
def __init__(self, constructor):
self.sampler = constructor(sd_model)
- self.orig_p_sample_ddim = self.sampler.p_sample_ddim
+ self.orig_p_sample_ddim = self.sampler.p_sample_ddim if hasattr(self.sampler, 'p_sample_ddim') else None
self.mask = None
self.nmask = None
self.init_latent = None