github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-18 02:41:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix crash (be a little less aggressive clearing out the attention slice)

Browse Source
This commit is contained in:
damian0815
2022-11-01 20:08:52 +01:00
committed by Lincoln Stein
parent 214d276379
commit 349cc25433
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -142,7 +142,7 @@ class InvokeAIDiffuserComponent:
for type in cross_attention_control_types_to_do:
CrossAttentionControl.request_save_attention_maps(self.model, type)
_ = self.model_forward_callback(x, sigma, conditioning)
CrossAttentionControl.clear_requests(self.model)
CrossAttentionControl.clear_requests(self.model, clear_attn_slice=False)
# process x again, using the saved attention maps to control where self.edited_conditioning will be applied
for type in cross_attention_control_types_to_do: