fix bug of Lama node

chflame163 · Jul 24, 2024 · dd264b5 · dd264b5
1 parent 502f3f2
commit dd264b5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/py/lama.py b/py/lama.py
@@ -91,7 +91,7 @@ def lama(self, image, mask, lama_model, device, invert_mask, mask_grow, mask_blu
                     log(f"Error: {NODE_NAME} skipped, because unable to create temporary file.", message_type='error')
                     return (image, )
                 # process
-                from iopaint import cli
+                from .iopaint import cli
                 cli.run(model=lama_model, device=device, image=Path(image_dir), mask=Path(mask_dir), output=Path(result_dir))
                 ret_image = check_image_file(os.path.join(result_dir, file_name), 500)
                 shutil.rmtree(temp_dir)

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui_layerstyle"
 description = "A set of nodes for ComfyUI it generate image like Adobe Photoshop's Layer Style. the Drop Shadow is first completed node, and follow-up work is in progress."
-version = "1.0.18"
+version = "1.0.19"
 license = "MIT"
 dependencies = ["numpy", "pillow", "torch", "matplotlib", "Scipy", "scikit_image", "opencv-contrib-python", "pymatting", "segment_anything", "timm", "addict", "yapf", "colour-science", "wget", "mediapipe", "loguru", "typer_config", "fastapi", "rich", "google-generativeai", "diffusers", "omegaconf", "tqdm", "transformers", "kornia", "image-reward", "ultralytics", "blend_modes", "blind-watermark", "qrcode", "pyzbar", "psd-tools"]