Skip to content

Commit 74c63f6

Browse files
authored
Update AILab_BiRefNet.py
1 parent 9fb0be1 commit 74c63f6

File tree

1 file changed

+29
-1
lines changed

1 file changed

+29
-1
lines changed

AILab_BiRefNet.py

Lines changed: 29 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,34 @@
141141
"default_res": 2048,
142142
"max_res": 2560,
143143
"min_res": 1024
144+
},
145+
"BiRefNet_dynamic": {
146+
"repo_id": "1038lab/BiRefNet_dynamic",
147+
"files": {
148+
"birefnet.py": "birefnet.py",
149+
"BiRefNet_config.py": "BiRefNet_config.py",
150+
"model.safetensors": "model.safetensors",
151+
"config.json": "config.json"
152+
},
153+
"cache_dir": "BiRefNet_dynamic",
154+
"description": "Dynamic model for high-resolution dichotomous image segmentation",
155+
"default_res": 1024,
156+
"max_res": 2048,
157+
"min_res": 512
158+
},
159+
"BiRefNet_lite-matting": {
160+
"repo_id": "1038lab/BiRefNet_lite-matting",
161+
"files": {
162+
"birefnet_lite.py": "birefnet_lite.py",
163+
"BiRefNet_config.py": "BiRefNet_config.py",
164+
"model.safetensors": "model.safetensors",
165+
"config.json": "config.json"
166+
},
167+
"cache_dir": "BiRefNet_lite-matting",
168+
"description": "Lightweight matting model for general purpose",
169+
"default_res": 1024,
170+
"max_res": 2048,
171+
"min_res": 512
144172
}
145173
}
146174

@@ -438,4 +466,4 @@ def hex_to_rgba(hex_color):
438466

439467
NODE_DISPLAY_NAME_MAPPINGS = {
440468
"BiRefNetRMBG": "BiRefNet (RMBG)"
441-
}
469+
}

0 commit comments

Comments
 (0)