fix: progress latent collection bug

brycedrennan · Jan 3, 2024 · 59206e6 · 59206e6
1 parent 5ad9c3c
commit 59206e6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/imaginairy/api/generate_refiners.py b/imaginairy/api/generate_refiners.py
@@ -336,7 +336,8 @@ def latent_logger(latents):
                     condition_scale=prompt.prompt_strength,
                     **text_conditioning_kwargs,
                 )
-                lc.progress_latent_callback(x)
+                if lc.progress_latent_callback:
+                    lc.progress_latent_callback(x)
             # trying to clear memory. not sure if this helps
             sd.unet.set_context(context="self_attention_map", value={})
             sd.unet._reset_context()