mlboydaisuke's picture
Add config.json so the Hub can count downloads
dadf281 verified
Raw
History Blame Contribute Delete
1.36 kB
{
"model_type": "executorch-pte",
"format": "pte",
"framework": "ExecuTorch",
"source": "facebook/sam2.1-hiera-tiny",
"inputs": [
[
1,
256,
64,
64
],
[
1,
32,
256,
256
],
[
1,
64,
128,
128
],
[
1,
1,
1,
2
],
[
1,
1,
1
]
],
"variants": [
{
"file": "sam21_tiny_decoder_coreml_all.pte",
"backend": "coreml",
"precision": "fp16",
"size_mb": 12.7
},
{
"file": "sam21_tiny_decoder_xnnpack_fp16.pte",
"backend": "xnnpack",
"precision": "fp16",
"size_mb": 12.6
},
{
"file": "sam21_tiny_decoder_xnnpack_fp32.pte",
"backend": "xnnpack",
"precision": "fp32",
"size_mb": 24.7
},
{
"file": "sam21_tiny_encoder_coreml_all.pte",
"backend": "coreml",
"precision": "fp16",
"size_mb": 70.0
},
{
"file": "sam21_tiny_encoder_xnnpack_fp16.pte",
"backend": "xnnpack",
"precision": "fp16",
"size_mb": 55.6
},
{
"file": "sam21_tiny_encoder_xnnpack_fp32.pte",
"backend": "xnnpack",
"precision": "fp32",
"size_mb": 109.2
}
],
"note": "ExecuTorch .pte exports. See README.md for variant selection and run instructions."
}