Fix PixArtT5TextEncode error

'tuple' object has no attribute 'squeeze'
2023-10-29 07:44:48 +05:00
parent 67ab30aa18
commit 0a176228f3
1 changed files with 2 additions and 1 deletions
@@ -137,7 +137,8 @@ class PixArtT5TextEncode:
 		masked_embs, keep_index = self.mask_feature(
 			embs.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)
 		return ([[masked_embs, {}]], )