Commit 18d6fe43 authored by C43H66N12O12S2's avatar C43H66N12O12S2 Committed by GitHub

.....

parent d63dbb3a
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment