use right value for step arg in img_callback in decode()

This commit is contained in:
Lincoln Stein 2022-10-03 05:47:32 -04:00
parent 7c6dbcb14a
commit 1ab09e7a06

View File

@ -346,7 +346,7 @@ class Sampler(object):
x_dec, pred_x0, e_t = outs
if img_callback:
img_callback(x_dec)
img_callback(x_dec,i)
return x_dec