diff options
| author | random_thoughtss <random_thoughtss@proton.me> | 2022-10-19 21:14:24 +0000 |
|---|---|---|
| committer | random_thoughtss <random_thoughtss@proton.me> | 2022-10-19 21:14:24 +0000 |
| commit | dde9f960727bfe151d418e43685a2881cf580a17 (patch) | |
| tree | aa750dc2b24581ac4162aaa1b332a36ecbf293e0 /modules/sd_samplers.py | |
| parent | 0719c10bf1b817364a498ee11b90d30d3d527344 (diff) | |
| download | stable-diffusion-webui-gfx803-dde9f960727bfe151d418e43685a2881cf580a17.tar.gz stable-diffusion-webui-gfx803-dde9f960727bfe151d418e43685a2881cf580a17.tar.bz2 stable-diffusion-webui-gfx803-dde9f960727bfe151d418e43685a2881cf580a17.zip | |
added support for ddim img2img
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 9d3cf289..d270e4df 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -208,6 +208,12 @@ class VanillaStableDiffusionSampler: self.init_latent = x
self.step = 0
+ # Wrap the conditioning models with additional image conditioning for inpainting model
+ if image_conditioning is not None:
+ conditioning = {"c_concat": [image_conditioning], "c_crossattn": [conditioning]}
+ unconditional_conditioning = {"c_concat": [image_conditioning], "c_crossattn": [unconditional_conditioning]}
+
+
samples = self.launch_sampling(steps, lambda: self.sampler.decode(x1, conditioning, t_enc, unconditional_guidance_scale=p.cfg_scale, unconditional_conditioning=unconditional_conditioning))
return samples
|