About the noise_pred calculation in sds loss in dual_branch_guidance.py

[silence-tang]

Hi, dear authors. The code from line 541 to line 544 in dual_branch_guidance.py is here:

noise_pred_text, noise_pred_uncond = noise_pred.chunk(2) noise_pred = noise_pred_text + self.cfg.guidance_scale * (noise_pred_text - noise_pred_uncond)

I wonder if there is a small mistake. It seems that the noise_pred should be calculated by noise_pred = noise_pred_uncond + self.cfg.guidance_scale * (noise_pred_text - noise_pred_uncond)?

I don't know if I miss something important. Looking forward to hearing from you!