From 32c525011bfcb47e796ff44c7e4b1d785c5b0273 Mon Sep 17 00:00:00 2001 From: Robin Rombach Date: Mon, 4 Apr 2022 22:31:35 +0200 Subject: [PATCH] unconditional guidance with general s!=1.0 Example: "negative weights" --- ldm/models/diffusion/ddim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/models/diffusion/ddim.py b/ldm/models/diffusion/ddim.py index cc51991..b8ff309 100644 --- a/ldm/models/diffusion/ddim.py +++ b/ldm/models/diffusion/ddim.py @@ -168,7 +168,7 @@ class DDIMSampler(object): b, *_, device = *x.shape, x.device e_t = self.model.apply_model(x, t, c) - if unconditional_guidance_scale > 1.: + if unconditional_guidance_scale != 1.: assert unconditional_conditioning is not None e_t_uncond = self.model.apply_model(x, t, unconditional_conditioning) e_t = e_t_uncond + unconditional_guidance_scale * (e_t - e_t_uncond)