From 783baba477d7b679012ffdf3cf3d384ac009a3f2 Mon Sep 17 00:00:00 2001
From: Asidert <612boris40@gmail.com>
Date: Mon, 22 Dec 2025 16:27:18 +0400
Subject: [PATCH] Import fix

---
 src/nodes.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/nodes.py b/src/nodes.py
index 7d0aa97..bea09f6 100644
--- a/src/nodes.py
+++ b/src/nodes.py
@@ -4,7 +4,7 @@ import numpy as np
 
 from PIL import Image
 from io import BytesIO
-from torch import Tensor
+import torch
 
 
 class LoadImageFromBase64:
@@ -24,7 +24,7 @@ class LoadImageFromBase64:
         code = cv2.COLOR_BGRA2RGB if image.ndim > 2 and image.shape[2] >= 4 else cv2.COLOR_BGR2RGB
         return cv2.cvtColor(image, code)
 
-    def main(self, data: str) -> tuple[Tensor, Tensor]:
+    def main(self, data: str) -> tuple[torch.Tensor, torch.Tensor]:
         nparr = np.frombuffer(base64.b64decode(data), np.uint8)
 
         result = cv2.imdecode(nparr, cv2.IMREAD_UNCHANGED)
@@ -58,7 +58,7 @@ class ConvertImageToBase64:
     OUTPUT_NODE = True
     CATEGORY = "image"
 
-    def main(self, images: Tensor) -> list[dict[str, str]]:
+    def main(self, images: torch.Tensor) -> list[dict[str, str]]:
         results: list[dict[str, str]] = list()
 
         for image in images: