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 )