revert to applying mask before denoising for k-diffusion, like it was before

This commit is contained in:
AUTOMATIC1111
2023-08-14 08:59:15 +03:00
parent cda2f0a162
commit c1a31ec9f7
2 changed files with 6 additions and 1 deletions

View File

@@ -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)