From 7c6dbcb14a75fd6cfceddf92c0336cfd4609b000 Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lincoln.stein@gmail.com>
Date: Mon, 3 Oct 2022 05:46:23 -0400
Subject: [PATCH] use right value for step arg in img_callback

---
 ldm/models/diffusion/sampler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ldm/models/diffusion/sampler.py b/ldm/models/diffusion/sampler.py
index 78196f8891..437b674518 100644
--- a/ldm/models/diffusion/sampler.py
+++ b/ldm/models/diffusion/sampler.py
@@ -272,7 +272,7 @@ class Sampler(object):
             if callback:
                 callback(i)
             if img_callback:
-                img_callback(img,step)
+                img_callback(img,i)
 
             if index % log_every_t == 0 or index == total_steps - 1:
                 intermediates['x_inter'].append(img)