dome272/Diffusion-Models-pytorch

CFG is right?

Edward0Liu opened this issue · 1 comments

In the paper, CLASSIFIER-FREE DIFFUSION GUIDANCE, epsilont = (1 + w)epsilonθ(zt, c) - wepsilonθ(zt).
But, codes's epsilont = lerp(epsilonθ(zt), epsilonθ(zt, c), w) = (1 - w)*epsilonθ(zt) + epsilonθ(zt, c)?
Am I right? Are those two the same?
Just a issue.

Yea they should be the same