mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-28 01:47:32 +08:00
Update model.py
This commit is contained in:
parent
893915b949
commit
c910e8bc1b
@ -141,7 +141,6 @@ class LensJointAttention(nn.Module):
|
|||||||
img_q, img_k, img_v = img_qkv.unbind(dim=2)
|
img_q, img_k, img_v = img_qkv.unbind(dim=2)
|
||||||
img_q = self.norm_q(img_q)
|
img_q = self.norm_q(img_q)
|
||||||
img_k = self.norm_k(img_k)
|
img_k = self.norm_k(img_k)
|
||||||
img_v = img_v.contiguous()
|
|
||||||
del img_qkv
|
del img_qkv
|
||||||
|
|
||||||
# text stream
|
# text stream
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user