RMBG-2.0 / config.json
OriLib's picture
Add birefnet model_type and tag (#22)
dea0983 verified
raw
history blame contribute delete
433 Bytes
{
"_name_or_path": "ZhengPeng7/BiRefNet",
"architectures": [
"BiRefNet"
],
"auto_map": {
"AutoConfig": "BiRefNet_config.BiRefNetConfig",
"AutoModelForImageSegmentation": "birefnet.BiRefNet"
},
"custom_pipelines": {
"image-segmentation": {
"pt": [
"AutoModelForImageSegmentation"
],
"tf": [],
"type": "image"
}
},
"bb_pretrained": false,
"model_type": "birefnet"
}