diff --git a/py/BiRefNet/gen_best_ep.py b/py/BiRefNet/gen_best_ep.py index c142835..6c2e4bd 100644 --- a/py/BiRefNet/gen_best_ep.py +++ b/py/BiRefNet/gen_best_ep.py @@ -2,7 +2,7 @@ import os from glob import glob import numpy as np -from BiRefNet..config import Config +from BiRefNet.config import Config config = Config() diff --git a/py/BiRefNet/models/backbones/build_backbone.py b/py/BiRefNet/models/backbones/build_backbone.py index 2507d15..699c6a8 100644 --- a/py/BiRefNet/models/backbones/build_backbone.py +++ b/py/BiRefNet/models/backbones/build_backbone.py @@ -2,9 +2,9 @@ import torch import torch.nn as nn from collections import OrderedDict from torchvision.models import vgg16, vgg16_bn, VGG16_Weights, VGG16_BN_Weights, resnet50, ResNet50_Weights -from models.backbones.pvt_v2 import pvt_v2_b0, pvt_v2_b1, pvt_v2_b2, pvt_v2_b5 -from models.backbones.swin_v1 import swin_v1_t, swin_v1_s, swin_v1_b, swin_v1_l -from config import Config +from BiRefNet.models.backbones.pvt_v2 import pvt_v2_b0, pvt_v2_b1, pvt_v2_b2, pvt_v2_b5 +from BiRefNet.models.backbones.swin_v1 import swin_v1_t, swin_v1_s, swin_v1_b, swin_v1_l +from BiRefNet.config import Config config = Config() diff --git a/py/BiRefNet/models/backbones/pvt_v2.py b/py/BiRefNet/models/backbones/pvt_v2.py index e80fc6f..ce6720b 100644 --- a/py/BiRefNet/models/backbones/pvt_v2.py +++ b/py/BiRefNet/models/backbones/pvt_v2.py @@ -7,7 +7,7 @@ from timm.models.registry import register_model import math -from config import Config +from BiRefNet.config import Config config = Config() diff --git a/py/BiRefNet/models/backbones/swin_v1.py b/py/BiRefNet/models/backbones/swin_v1.py index 45045b8..b3a9396 100644 --- a/py/BiRefNet/models/backbones/swin_v1.py +++ b/py/BiRefNet/models/backbones/swin_v1.py @@ -12,7 +12,7 @@ import torch.utils.checkpoint as checkpoint import numpy as np from timm.models.layers import DropPath, to_2tuple, trunc_normal_ -from config import Config +from BiRefNet.config import Config config = Config() diff --git a/pyproject.toml b/pyproject.toml index 2f0aa73..3280189 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.47" +version = "1.0.48" 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", "inference-cli[yolo-world]", "inference-gpu[yolo-world]", "onnxruntime"]