From d77d724d664f74486984bfcc247bb9cb42b72e28 Mon Sep 17 00:00:00 2001 From: Luo Chaofan <79003314+fkcptlst@users.noreply.github.com> Date: Sun, 7 Jan 2024 22:11:35 +0800 Subject: [PATCH] fixed guidance in stable_diffusion.py --- guidance/stable_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/guidance/stable_diffusion.py b/guidance/stable_diffusion.py index 02cb1e1..c992091 100644 --- a/guidance/stable_diffusion.py +++ b/guidance/stable_diffusion.py @@ -229,7 +229,7 @@ def compute_grad_sds( # perform guidance (high scale from paper!) noise_pred_text, noise_pred_uncond = noise_pred.chunk(2) - noise_pred = noise_pred_text + self.cfg.guidance_scale * ( + noise_pred = noise_pred_uncond + self.cfg.guidance_scale * ( noise_pred_text - noise_pred_uncond )