Diffusion short course bug

helper function: perturbs an image to a specified noise level

def perturb_input(x, t, noise):
return ab_t.sqrt()[t, None, None, None] * x + (1 - ab_t[t, None, None, None]) * noise

I’m pretty sure it should be 1-ab_t.sqrt(). Submitting here because the website won’t let me create course q&a.

You have posted in the AI Questions area. But your message seems more appropriate for the forum area for that specific course.

You can use the “pencil” icon in the thread title to move your post to the appropriate area. It is more likely to be noticed by a mentor or staff for that course.