Commit 76a616fa authored by C43H66N12O12S2's avatar C43H66N12O12S2 Committed by GitHub

Update sd_hijack_optimizations.py

parent 5d54f35c
...@@ -176,7 +176,7 @@ def cross_attention_attnblock_forward(self, x): ...@@ -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()
......
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