aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_timesteps.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-14 05:59:15 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2023-08-14 05:59:15 +0000
commitc1a31ec9f75c8dfe4ddcb0061f06e2704db98359 (patch)
tree4d2c238cee27de8ac90a4db74325bc3d2d25d861 /modules/sd_samplers_timesteps.py
parentcda2f0a1620c3b49bb3408c30796160ed29bc87d (diff)
downloadstable-diffusion-webui-gfx803-c1a31ec9f75c8dfe4ddcb0061f06e2704db98359.tar.gz
stable-diffusion-webui-gfx803-c1a31ec9f75c8dfe4ddcb0061f06e2704db98359.tar.bz2
stable-diffusion-webui-gfx803-c1a31ec9f75c8dfe4ddcb0061f06e2704db98359.zip
revert to applying mask before denoising for k-diffusion, like it was before
Diffstat (limited to 'modules/sd_samplers_timesteps.py')
-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 6aed2974..c1f534ed 100644
--- a/modules/sd_samplers_timesteps.py
+++ b/modules/sd_samplers_timesteps.py
@@ -49,6 +49,7 @@ class CFGDenoiserTimesteps(CFGDenoiser):
super().__init__(sampler)
self.alphas = shared.sd_model.alphas_cumprod
+ self.mask_before_denoising = True
def get_pred_x0(self, x_in, x_out, sigma):
ts = sigma.to(dtype=int)