Commit 57f29bd6 authored by CodeHatchling's avatar CodeHatchling
Browse files

Re-introduce latent blending step from the vanilla inpainting procedure.

parent 1455159c
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -1597,6 +1597,9 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing):

        samples = self.sampler.sample_img2img(self, self.init_latent, x, conditioning, unconditional_conditioning, image_conditioning=self.image_conditioning)

        if self.mask is not None and self.soft_inpainting is None:
            samples = samples * self.nmask + self.init_latent * self.mask

        del x
        devices.torch_gc()