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

fix a bug that broke cross attention control index mapping

Browse Source
This commit is contained in:
damian0815 2022-11-02 00:33:00 +01:00 committed by blessedcoolant
parent 4513320bf1
commit 688d7258f1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/models/diffusion/cross_attention_control.py
View File

@ -68,6 +68,8 @@ class CrossAttentionControl:
indices[b0:b1] = indices_target[a0:a1] indices[b0:b1] = indices_target[a0:a1]
mask[b0:b1] = 1 mask[b0:b1] = 1
cls.inject_attention_function(model)
for m in cls.get_attention_modules(model, cls.CrossAttentionType.SELF): for m in cls.get_attention_modules(model, cls.CrossAttentionType.SELF):
m.last_attn_slice_mask = None m.last_attn_slice_mask = None
m.last_attn_slice_indices = None m.last_attn_slice_indices = None
@ -76,8 +78,6 @@ class CrossAttentionControl:
m.last_attn_slice_mask = mask.to(device) m.last_attn_slice_mask = mask.to(device)
m.last_attn_slice_indices = indices.to(device) m.last_attn_slice_indices = indices.to(device)
cls.inject_attention_function(model)
class CrossAttentionType(Enum): class CrossAttentionType(Enum):
SELF = 1 SELF = 1