fix typo which caused crash in sampler.py

This commit is contained in:
Lincoln Stein 2022-10-02 22:31:11 -04:00
parent 8ba5e385ec
commit 5a88be3744

View File

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