nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Temp fix for mask shape

Browse Source
This commit is contained in:
City
2023-12-08 17:23:51 +01:00
parent a634b1ea09
commit 80d5d9299b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/models/PixArt_blocks.py
+1 -1
View File
@@ -73,7 +73,7 @@ class MultiHeadCrossAttention(nn.Module):
q, k, v = map(lambda t: t.permute(0, 2, 1, 3),(q, k, v),) q, k, v = map(lambda t: t.permute(0, 2, 1, 3),(q, k, v),)
attn_mask = None attn_mask = None
if mask is not None: if mask is not None and len(mask) > 1:
# This is probably wrong # This is probably wrong
attn_mask = torch.zeros( attn_mask = torch.zeros(
[1, q.shape[1], q.shape[2], v.shape[2]], [1, q.shape[1], q.shape[2], v.shape[2]],