Fix spread

2024-09-05 20:16:52 +03:00
parent c5a05e4455
commit 41d0a72e1f
1 changed files with 2 additions and 2 deletions
@@ -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"
-}
+}