Update nodes.py

2024-04-09 19:37:33 +03:00
parent 194b01b3da
commit 835020fa7b
1 changed files with 2 additions and 2 deletions
@@ -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,)