From 835020fa7b06c0608fbfcfd8aedb19d2fedf1b7f Mon Sep 17 00:00:00 2001
From: Kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 9 Apr 2024 19:37:33 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index 51f4f2a..e8b5c66 100644
--- a/nodes.py
+++ b/nodes.py
@@ -262,7 +262,7 @@ class ella_model_loader:
             print("pipeline created")
             pbar.update(1)
             pipe.unet = ella_unet
-            t5_encoder = T5TextEmbedder().to(pipe.device, dtype=torch.float16)
+            t5_encoder = T5TextEmbedder().to(pipe.device, dtype=dtype)
             ella_model = {
                 'pipe': pipe,
                 'ella': ella,
@@ -381,7 +381,7 @@ class ella_sampler:
             ],
                 output_type="np.array",
             ).images
-            print(images.shape)
+
             tensor = torch.from_numpy(images).cpu().float()
          
             return (tensor,)