Commit f9549d1c authored by random_thoughtss's avatar random_thoughtss
Browse files

Added option to use unmasked conditioning image.

parent 3e15f8e0

modules/processing.py

+5 −1
Original line number Diff line number Diff line
@@ -768,7 +768,11 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing): 


            # Create another latent image, this time with a masked version of the original input.

            conditioning_mask = conditioning_mask.to(image.device)

            conditioning_image = image * (1.0 - conditioning_mask)



            conditioning_image = image

            if shared.opts.inpainting_mask_image:

                conditioning_image = conditioning_image * (1.0 - conditioning_mask)                

            

            conditioning_image = self.sd_model.get_first_stage_encoding(self.sd_model.encode_first_stage(conditioning_image))



            # Create the concatenated conditioning tensor to be fed to `c_concat`

modules/shared.py

+1 −0
Original line number Diff line number Diff line
@@ -320,6 +320,7 @@ options_templates.update(options_section(('sampler-params', "Sampler parameters" 
    's_tmin':  OptionInfo(0.0, "sigma tmin",  gr.Slider, {"minimum": 0.0, "maximum": 1.0, "step": 0.01}),

    's_noise': OptionInfo(1.0, "sigma noise", gr.Slider, {"minimum": 0.0, "maximum": 1.0, "step": 0.01}),

    'eta_noise_seed_delta': OptionInfo(0, "Eta noise seed delta", gr.Number, {"precision": 0}),

    "inpainting_mask_image": OptionInfo(True, "Mask original image for conditioning used by inpainting model."),

}))

CRA Git | Maintained and supported by SUSTech CRA and CCSE