Update sd_hijack_optimizations.py

This commit is contained in:
C43H66N12O12S2 2022-10-08 11:55:38 +03:00 committed by GitHub
parent 5d54f35c58
commit 76a616fa6b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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()