fix: progress latent collection bug

4 months ago · d148bc1537
parent fdb48399af
commit d148bc1537
1 changed files with 2 additions and 1 deletions
--- a/imaginairy/api/generate_refiners.py
+++ b/imaginairy/api/generate_refiners.py
@ -336,7 +336,8 @@ def generate_single_image(
                    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()