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

Fix PixArtT5TextEncode error

Browse Source 
'tuple' object has no attribute 'squeeze'
This commit is contained in:
deinferno
2023-10-29 07:44:48 +05:00
parent 67ab30aa18
commit 0a176228f3
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/nodes.py
+2 -1
View File
@@ -137,7 +137,8 @@ class PixArtT5TextEncode:
masked_embs, keep_index = self.mask_feature( masked_embs, keep_index = self.mask_feature(
embs.detach().to("cpu"), embs.detach().to("cpu"),
mask.detach().to("cpu") mask.detach().to("cpu")
).squeeze(0) # match CLIP/internal )
masked_embs = masked_embs.squeeze(0) # match CLIP/internal
print("Encoded T5:", masked_embs.shape) print("Encoded T5:", masked_embs.shape)
return ([[masked_embs, {}]], ) return ([[masked_embs, {}]], )