nodes/1038lab-ComfyUI-RMBG
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update AILab_BiRefNet.py

Browse Source
This commit is contained in:
AI Lab
2025-07-07 20:52:39 -07:00
committed by GitHub
parent 9fb0be165d
commit 74c63f6123
1 changed files with 29 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
AILab_BiRefNet.py
+29 -1
View File
@@ -141,6 +141,34 @@ MODEL_CONFIG = {
"default_res": 2048,
"max_res": 2560,
"min_res": 1024
},
"BiRefNet_dynamic": {
"repo_id": "1038lab/BiRefNet_dynamic",
"files": {
"birefnet.py": "birefnet.py",
"BiRefNet_config.py": "BiRefNet_config.py",
"model.safetensors": "model.safetensors",
"config.json": "config.json"
},
"cache_dir": "BiRefNet_dynamic",
"description": "Dynamic model for high-resolution dichotomous image segmentation",
"default_res": 1024,
"max_res": 2048,
"min_res": 512
},
"BiRefNet_lite-matting": {
"repo_id": "1038lab/BiRefNet_lite-matting",
"files": {
"birefnet_lite.py": "birefnet_lite.py",
"BiRefNet_config.py": "BiRefNet_config.py",
"model.safetensors": "model.safetensors",
"config.json": "config.json"
},
"cache_dir": "BiRefNet_lite-matting",
"description": "Lightweight matting model for general purpose",
"default_res": 1024,
"max_res": 2048,
"min_res": 512
}
}
@@ -438,4 +466,4 @@ NODE_CLASS_MAPPINGS = {
NODE_DISPLAY_NAME_MAPPINGS = {
"BiRefNetRMBG": "BiRefNet (RMBG)"
}
}