Update sd_hijack_optimizations.py
This commit is contained in:
parent
5d54f35c58
commit
76a616fa6b
1 changed files with 1 additions and 1 deletions
|
@ -176,7 +176,7 @@ def cross_attention_attnblock_forward(self, x):
|
||||||
|
|
||||||
return h3
|
return h3
|
||||||
|
|
||||||
def xformers_attnblock_forward(self, x):
|
def xformers_attnblock_forward(self, x):
|
||||||
h_ = x
|
h_ = x
|
||||||
h_ = self.norm(h_)
|
h_ = self.norm(h_)
|
||||||
q1 = self.q(h_).contiguous()
|
q1 = self.q(h_).contiguous()
|
||||||
|
|
Loading…
Reference in a new issue