github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-17 22:51:56 -05: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
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]
mask[b0:b1] = 1
cls.inject_attention_function(model)
for m in cls.get_attention_modules(model, cls.CrossAttentionType.SELF):
m.last_attn_slice_mask = 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_indices = indices.to(device)
cls.inject_attention_function(model)
class CrossAttentionType(Enum):
SELF = 1