diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-21 06:00:39 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-21 06:00:39 +0000 |
| commit | 45872181902ada06267e2de601586d512cf5df1a (patch) | |
| tree | 307b0560319740a196db74ac748f795e51232861 /modules/sd_hijack_inpainting.py | |
| parent | 49533eed9e3aad19e9868ee140708baec4fd44be (diff) | |
| download | stable-diffusion-webui-gfx803-45872181902ada06267e2de601586d512cf5df1a.tar.gz stable-diffusion-webui-gfx803-45872181902ada06267e2de601586d512cf5df1a.tar.bz2 stable-diffusion-webui-gfx803-45872181902ada06267e2de601586d512cf5df1a.zip | |
updated readme and some small stylistic changes to code
Diffstat (limited to 'modules/sd_hijack_inpainting.py')
| -rw-r--r-- | modules/sd_hijack_inpainting.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_hijack_inpainting.py b/modules/sd_hijack_inpainting.py index 43938071..fd92a335 100644 --- a/modules/sd_hijack_inpainting.py +++ b/modules/sd_hijack_inpainting.py @@ -301,6 +301,7 @@ def get_unconditional_conditioning(self, batch_size, null_label=None): c = repeat(c, "1 ... -> b ...", b=batch_size).to(self.device) return c + class LatentInpaintDiffusion(LatentDiffusion): def __init__( self, @@ -314,9 +315,11 @@ class LatentInpaintDiffusion(LatentDiffusion): assert self.masked_image_key in concat_keys self.concat_keys = concat_keys + def should_hijack_inpainting(checkpoint_info): return str(checkpoint_info.filename).endswith("inpainting.ckpt") and not checkpoint_info.config.endswith("inpainting.yaml") + def do_inpainting_hijack(): ldm.models.diffusion.ddpm.get_unconditional_conditioning = get_unconditional_conditioning ldm.models.diffusion.ddpm.LatentInpaintDiffusion = LatentInpaintDiffusion |