mlboydaisuke's picture
Add config.json so the Hub can count downloads
87fb708 verified
Raw
History Blame Contribute Delete
535 Bytes
{
"model_type": "executorch-pte",
"format": "pte",
"framework": "ExecuTorch",
"source": "torchvision raft_small (C_T_V2, FlyingChairs + FlyingThings3D)",
"inputs": [
[
1,
3,
384,
512
],
[
1,
3,
384,
512
]
],
"variants": [
{
"file": "raft_small_xnnpack_fp32.pte",
"backend": "xnnpack",
"precision": "fp32",
"size_mb": 4.4
}
],
"note": "ExecuTorch .pte exports. See README.md for variant selection and run instructions."
}