BiRefNet / config.json
ZhengPeng7's picture
Ensemble BiRefNet codes to HF to load model with weights in one line.
40a7dcf
raw
history blame
413 Bytes
{
"_name_or_path": "ZhengPeng7/BiRefNet-w_codes",
"architectures": [
"BiRefNet"
],
"auto_map": {
"AutoConfig": "BiRefNet_config.BiRefNetConfig",
"AutoModelForImageSegmentation": "birefnet.BiRefNet"
},
"custom_pipelines": {
"image-segmentation": {
"pt": [
"AutoModelForImageSegmentation"
],
"tf": [],
"type": "image"
}
},
"bb_pretrained": false
}