fix model loading bug of SegmentAnythingUltraV2 node

chflame163 · Nov 2, 2024 · ad8fee9 · ad8fee9
1 parent db05355
commit ad8fee9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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"]