aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_hijack.py
diff options
context:
space:
mode:
authorC43H66N12O12S2 <36072735+C43H66N12O12S2@users.noreply.github.com>2022-09-17 22:21:50 +0000
committerGitHub <noreply@github.com>2022-09-17 22:21:50 +0000
commit18d6fe4346e2543522cd2a64c71207e45632a46b (patch)
treecc8622bb42ef7a7d055b70cc71385789e8c8bef4 /modules/sd_hijack.py
parentd63dbb3acc16a737711fddb3a954238f82deb2fa (diff)
downloadstable-diffusion-webui-gfx803-18d6fe4346e2543522cd2a64c71207e45632a46b.tar.gz
stable-diffusion-webui-gfx803-18d6fe4346e2543522cd2a64c71207e45632a46b.tar.bz2
stable-diffusion-webui-gfx803-18d6fe4346e2543522cd2a64c71207e45632a46b.zip
.....
Diffstat (limited to 'modules/sd_hijack.py')
-rw-r--r--modules/sd_hijack.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index d819c01c..c4450ce4 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