fix bug of EVF-SAMUltra duplicate module name

chflame163 · Aug 29, 2024 · bb43b17 · bb43b17
1 parent 6414c85
commit bb43b17
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/py/evf_sam/evf_sam_inference.py b/py/evf_sam/evf_sam_inference.py
@@ -8,7 +8,8 @@
 from torchvision import transforms
 from torchvision.transforms.functional import InterpolationMode
 from transformers import AutoTokenizer, BitsAndBytesConfig
-from model.segment_anything.utils.transforms import ResizeLongestSide
+sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
+from .model.segment_anything.utils.transforms import ResizeLongestSide
 
 def sam_preprocess(
     x: np.ndarray,

diff --git a/py/evf_sam/model/__init__.py b/py/evf_sam/model/__init__.py
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.35"
+version = "1.0.36"
 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", "transparent-background", "huggingface_hub", "accelerate", "bitsandbytes", "torchscale", "wandb", "hydra-core", "psd-tools"]