diff --git a/BiRefNet_node.py b/BiRefNet_node.py index 144ce9f..e488254 100644 --- a/BiRefNet_node.py +++ b/BiRefNet_node.py @@ -17,7 +17,7 @@ from loguru import logger from folder_paths import models_dir -from utils import check_download_model +from downloader import check_download_model config = Config() diff --git a/utils.py b/downloader.py similarity index 100% rename from utils.py rename to downloader.py diff --git a/models/backbones/build_backbone.py b/models/backbones/build_backbone.py index ed1e6ea..92fa84a 100644 --- a/models/backbones/build_backbone.py +++ b/models/backbones/build_backbone.py @@ -5,7 +5,7 @@ from models.backbones.pvt_v2 import 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 utils import check_download_model +from downloader import check_download_model config = Config()