From 0a176228f3457a90d0f2459ecca196425fa05f14 Mon Sep 17 00:00:00 2001
From: deinferno <14363193+deinferno@users.noreply.github.com>
Date: Sun, 29 Oct 2023 07:40:40 +0500
Subject: [PATCH] Fix PixArtT5TextEncode error 'tuple' object has no attribute
 'squeeze'

---
 PixArt/nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/PixArt/nodes.py b/PixArt/nodes.py
index 76f165b..7292e8f 100644
--- a/PixArt/nodes.py
+++ b/PixArt/nodes.py
@@ -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, {}]], )