From 41d0a72e1f2d48ab29b737f1ccc268cabf42d03b Mon Sep 17 00:00:00 2001
From: asagi4 <130366179+asagi4@users.noreply.github.com>
Date: Wed, 4 Sep 2024 23:28:58 +0300
Subject: [PATCH] Fix spread

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

diff --git a/py/lama.py b/py/lama.py
index 03b265d..4d3f8d7 100644
--- a/py/lama.py
+++ b/py/lama.py
@@ -67,7 +67,7 @@ class LaMa:
                     _mask = tensor2pil(expand_mask(image2mask(_mask), mask_grow, mask_blur))
                 
                 ret_image = pixel_spread(tensor2pil(_image).convert('RGB'), ImageChops.invert(_mask.convert('RGB')))
-                ret_images.append(ret_image)
+                ret_images.append(pil2tensor(ret_image))
         else:
             temp_dir = os.path.join(folder_paths.get_temp_directory(), generate_random_name('_lama_', '_temp', 16))
             if os.path.isdir(temp_dir):
@@ -122,4 +122,4 @@ NODE_CLASS_MAPPINGS = {
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "LayerUtility: LaMa": "LayerUtility: LaMa"
-}
\ No newline at end of file
+}