better naming

Author: patrickvonplaten

examples/community/composable_stable_diffusion.py

@@ -562,8 +562,8 @@ def __call__(
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
                     progress_bar.update()
                     if callback is not None and i % callback_steps == 0:
-                        step_idx = i // getattr("scheduler", "order", 1)
-                        callback(step_idx, t, latents)
+                        callback_idx = i // getattr("scheduler", "order", 1)
+                        callback(callback_idx, t, latents)
 
         # 8. Post-processing
         image = self.decode_latents(latents)

examples/community/img2img_inpainting.py

@@ -434,8 +434,8 @@ def __call__(
 
             # call the callback, if provided
             if callback is not None and i % callback_steps == 0:
-                step_idx = i // getattr("scheduler", "order", 1)
-                callback(step_idx, t, latents)
+                callback_idx = i // getattr("scheduler", "order", 1)
+                callback(callback_idx, t, latents)
 
         latents = 1 / 0.18215 * latents
         image = self.vae.decode(latents).sample

examples/community/interpolate_stable_diffusion.py

@@ -372,8 +372,8 @@ def __call__(
 
             # call the callback, if provided
             if callback is not None and i % callback_steps == 0:
-                step_idx = i // getattr("scheduler", "order", 1)
-                callback(step_idx, t, latents)
+                callback_idx = i // getattr("scheduler", "order", 1)
+                callback(callback_idx, t, latents)
 
         latents = 1 / 0.18215 * latents
         image = self.vae.decode(latents).sample

examples/community/lpw_stable_diffusion.py

@@ -1088,8 +1088,8 @@ def __call__(
                     progress_bar.update()
                     if i % callback_steps == 0:
                         if callback is not None:
-                            step_idx = i // getattr("scheduler", "order", 1)
-                            callback(step_idx, t, latents)
+                            callback_idx = i // getattr("scheduler", "order", 1)
+                            callback(callback_idx, t, latents)
                         if is_cancelled_callback is not None and is_cancelled_callback():
                             return None
 

examples/community/lpw_stable_diffusion_onnx.py

@@ -846,8 +846,8 @@ def __call__(
             # call the callback, if provided
             if i % callback_steps == 0:
                 if callback is not None:
-                    step_idx = i // getattr("scheduler", "order", 1)
-                    callback(step_idx, t, latents)
+                    callback_idx = i // getattr("scheduler", "order", 1)
+                    callback(callback_idx, t, latents)
                 if is_cancelled_callback is not None and is_cancelled_callback():
                     return None
 

examples/community/lpw_stable_diffusion_xl.py

@@ -1182,8 +1182,8 @@ def __call__(
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
                     progress_bar.update()
                     if callback is not None and i % callback_steps == 0:
-                        step_idx = i // getattr("scheduler", "order", 1)
-                        callback(step_idx, t, latents)
+                        callback_idx = i // getattr("scheduler", "order", 1)
+                        callback(callback_idx, t, latents)
 
         if not output_type == "latent":
             # make sure the VAE is in float32 mode, as it overflows in float16

examples/community/masked_stable_diffusion_img2img.py

@@ -202,8 +202,8 @@ def __call__(
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
                     progress_bar.update()
                     if callback is not None and i % callback_steps == 0:
-                        step_idx = i // getattr("scheduler", "order", 1)
-                        callback(step_idx, t, latents)
+                        callback_idx = i // getattr("scheduler", "order", 1)
+                        callback(callback_idx, t, latents)
 
         if not output_type == "latent":
             scaled = latents / self.vae.config.scaling_factor

examples/community/multilingual_stable_diffusion.py

@@ -407,8 +407,8 @@ def __call__(
 
             # call the callback, if provided
             if callback is not None and i % callback_steps == 0:
-                step_idx = i // getattr("scheduler", "order", 1)
-                callback(step_idx, t, latents)
+                callback_idx = i // getattr("scheduler", "order", 1)
+                callback(callback_idx, t, latents)
 
         latents = 1 / 0.18215 * latents
         image = self.vae.decode(latents).sample

examples/community/pipeline_prompt2prompt.py

@@ -254,8 +254,8 @@ def __call__(
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
                     progress_bar.update()
                     if callback is not None and i % callback_steps == 0:
-                        step_idx = i // getattr("scheduler", "order", 1)
-                        callback(step_idx, t, latents)
+                        callback_idx = i // getattr("scheduler", "order", 1)
+                        callback(callback_idx, t, latents)
 
         # 8. Post-processing
         if not output_type == "latent":

examples/community/pipeline_zero1to3.py

@@ -865,8 +865,8 @@ def __call__(
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
                     progress_bar.update()
                     if callback is not None and i % callback_steps == 0:
-                        step_idx = i // getattr("scheduler", "order", 1)
-                        callback(step_idx, t, latents)
+                        callback_idx = i // getattr("scheduler", "order", 1)
+                        callback(callback_idx, t, latents)
 
         # 8. Post-processing
         has_nsfw_concept = None