Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

don't restore None

Browse Source
This commit is contained in:
damian 2023-01-26 17:44:27 +01:00
parent fb312f9ed3
commit c381788ab9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/models/diffusion/shared_invokeai_diffusion.py
View File

@ -69,7 +69,8 @@ class InvokeAIDiffuserComponent:
try:
yield None
finally:
self.remove_cross_attention_control(old_attn_processor)
if old_attn_processor is not None:
self.remove_cross_attention_control(old_attn_processor)
# TODO resuscitate attention map saving
#self.remove_attention_map_saving()