From fb312f9ed3bc2fd75942a7f6c877e1c5bcead7b5 Mon Sep 17 00:00:00 2001 From: damian Date: Thu, 26 Jan 2023 17:30:23 +0100 Subject: [PATCH] use the correct value - whoops --- ldm/models/diffusion/cross_attention_control.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/models/diffusion/cross_attention_control.py b/ldm/models/diffusion/cross_attention_control.py index 99ef1d49bc..770c71f110 100644 --- a/ldm/models/diffusion/cross_attention_control.py +++ b/ldm/models/diffusion/cross_attention_control.py @@ -636,7 +636,7 @@ class SlicedSwapCrossAttnProcesser(SlicedAttnProcessor): del remapped_original_attn_slice, modified_attn_slice - attn_slice = torch.bmm(attn_slice, original_value[start_idx:end_idx]) + attn_slice = torch.bmm(attn_slice, modified_value[start_idx:end_idx]) hidden_states[start_idx:end_idx] = attn_slice