MODNet-ExecuTorch / config.json
mlboydaisuke's picture
Add config.json so the Hub can count downloads
ffab07d verified
Raw
History Blame Contribute Delete
780 Bytes
{
"model_type": "executorch-pte",
"format": "pte",
"framework": "ExecuTorch",
"source": "ZHKKKe/MODNet + DavG25/modnet-pretrained-models ckpt",
"inputs": [
[
1,
3,
512,
512
]
],
"variants": [
{
"file": "modnet_portrait_matting_coreml_all.pte",
"backend": "coreml",
"precision": "fp16",
"size_mb": 13.8
},
{
"file": "modnet_portrait_matting_xnnpack_fp16.pte",
"backend": "xnnpack",
"precision": "fp16",
"size_mb": 24.4
},
{
"file": "modnet_portrait_matting_xnnpack_fp32.pte",
"backend": "xnnpack",
"precision": "fp32",
"size_mb": 26.1
}
],
"note": "ExecuTorch .pte exports. See README.md for variant selection and run instructions."
}