diff --git a/ldm/models/diffusion/sampler.py b/ldm/models/diffusion/sampler.py
index 437b674518..b62278c719 100644
--- a/ldm/models/diffusion/sampler.py
+++ b/ldm/models/diffusion/sampler.py
@@ -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