Skip to content

Commit

Permalink
fix EVF-SAMUltra moudle name
Browse files Browse the repository at this point in the history
  • Loading branch information
chflame163 committed Aug 29, 2024
1 parent 2c5c515 commit 5218591
Show file tree
Hide file tree
Showing 75 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions py/evf_sam/evf_sam_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from torchvision import transforms
from torchvision.transforms.functional import InterpolationMode
from transformers import AutoTokenizer, BitsAndBytesConfig
from .model.segment_anything.utils.transforms import ResizeLongestSide
from evf_sam_model.segment_anything.utils.transforms import ResizeLongestSide

def sam_preprocess(
x: np.ndarray,
Expand Down Expand Up @@ -93,17 +93,17 @@ def init_models(model_path:str, model_type:str, precision:str, load_in_bit:int=1
)

if model_type=="ori":
from model.evf_sam import EvfSamModel
from evf_sam_model.evf_sam import EvfSamModel
model = EvfSamModel.from_pretrained(
model_path, low_cpu_mem_usage=True, **kwargs
)
elif model_type=="effi":
from model.evf_effisam import EvfEffiSamModel
from evf_sam_model.evf_effisam import EvfEffiSamModel
model = EvfEffiSamModel.from_pretrained(
model_path, low_cpu_mem_usage=True, **kwargs
)
elif model_type=="sam2":
from model.evf_sam2 import EvfSam2Model
from evf_sam_model.evf_sam2 import EvfSam2Model
model = EvfSam2Model.from_pretrained(
model_path, low_cpu_mem_usage=True, **kwargs
)
Expand Down
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -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"]

Expand Down

0 comments on commit 5218591

Please sign in to comment.