From ad8fee9f170657e7b6f8fc9763d6798f33d8c4df Mon Sep 17 00:00:00 2001 From: chflame163 Date: Sat, 2 Nov 2024 12:44:19 +0800 Subject: [PATCH] fix model loading bug of SegmentAnythingUltraV2 node --- py/segment_anything_ultra_v2.py | 4 ++-- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/py/segment_anything_ultra_v2.py b/py/segment_anything_ultra_v2.py index d60e7b8..2e59874 100644 --- a/py/segment_anything_ultra_v2.py +++ b/py/segment_anything_ultra_v2.py @@ -63,8 +63,8 @@ def segment_anything_ultra_v2(self, image, sam_model, grounding_dino_model, thre self.DINO_MODEL = load_groundingdino_model(grounding_dino_model) self.previous_dino_model = grounding_dino_model - SAM_MODEL = load_sam_model(sam_model) - DINO_MODEL = load_groundingdino_model(grounding_dino_model) + # SAM_MODEL = load_sam_model(sam_model) + # DINO_MODEL = load_groundingdino_model(grounding_dino_model) ret_images = [] ret_masks = [] diff --git a/pyproject.toml b/pyproject.toml index 9c751c2..5a29ee3 100644 --- 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.86" +version = "1.0.87" license = "MIT" dependencies = ["numpy", "pillow", "torch", "matplotlib", "Scipy", "scikit_image", "scikit_learn", "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", "transparent-background", "huggingface_hub", "accelerate", "bitsandbytes", "torchscale", "wandb", "hydra-core", "psd-tools", "inference-cli[yolo-world]", "inference-gpu[yolo-world]", "onnxruntime", "peft", "iopath"]