diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index d819c01c6..c4450ce43 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -20,7 +20,7 @@ def split_cross_attention_forward_v1(self, x, context=None, mask=None): q = self.to_q(x) context = default(context, x) - k = self.to_k(context) * self.scale + k = self.to_k(context) v = self.to_v(context) del context, x @@ -50,7 +50,7 @@ def split_cross_attention_forward(self, x, context=None, mask=None): q_in = self.to_q(x) context = default(context, x) - k_in = self.to_k(context) + k_in = self.to_k(context) * self.scale v_in = self.to_v(context) del context, x