Archives/imaginAIry
2
0
Fork 0
mirror of https://github.com/brycedrennan/imaginAIry synced 2024-10-31 03:20:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: progress latent collection bug

Browse Source
This commit is contained in:
Bryce 2024-01-03 09:00:02 -08:00 committed by Bryce Drennan
parent fdb48399af
commit d148bc1537
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
imaginairy/api/generate_refiners.py
View File

@ -336,7 +336,8 @@ def generate_single_image(
condition_scale=prompt.prompt_strength, condition_scale=prompt.prompt_strength,
**text_conditioning_kwargs, **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 # trying to clear memory. not sure if this helps
sd.unet.set_context(context="self_attention_map", value={}) sd.unet.set_context(context="self_attention_map", value={})
sd.unet._reset_context() sd.unet._reset_context()