fix bug of BiRefNetUltraV2

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()

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()

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()
 

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()

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"]