Merge pull request #1 from deinferno/main

Fix PixArtT5TextEncode error
2023-10-29 12:24:12 +01:00
parent 67ab30aa18 0a176228f3
commit 1d5f745bbf
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, {}]], )