From 1cd5fd0c23f7c364be16915e8ac6e9189c074bfe Mon Sep 17 00:00:00 2001
From: nullquant <merlinkun@yandex.ru>
Date: Mon, 3 Jun 2024 11:17:15 +0300
Subject: [PATCH] Terminal node

---
 __init__.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index 59dfe92..8f3959d 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,5 +1,6 @@
 from .brushnet_nodes import BrushNetLoader, BrushNet, BlendInpaint, PowerPaintCLIPLoader, PowerPaint, CutForInpaint
 from .raunet_nodes import RAUNet
+import torch
 from subprocess import getoutput
 
 """
@@ -19,8 +20,8 @@ class Terminal:
                 }
     
     CATEGORY = "Tools"
-    RETURN_TYPES = ("STRING", )
-    RETURN_NAMES = ("text", )
+    RETURN_TYPES = ("IMAGE", )
+    RETURN_NAMES = ("image", )
     OUTPUT_NODE = True
 
     FUNCTION = "execute"
@@ -31,7 +32,7 @@ class Terminal:
             print(out)
         else:
             exec(f"{text}")
-        return (text, )
+        return (torch.zeros(1, 128, 128, 4), )