config.json

{
  "model_type": "executorch-pte",
  "format": "pte",
  "framework": "ExecuTorch",
  "source": "thohemp/6DRepNet + osanseviero/6DRepNet_300W_LP_AFLW2000 weights",
  "inputs": [
    [
      1,
      3,
      224,
      224
    ]
  ],
  "variants": [
    {
      "file": "sixdrepnet_headpose_coreml_all.pte",
      "backend": "coreml",
      "precision": "fp16",
      "size_mb": 78.8
    },
    {
      "file": "sixdrepnet_headpose_xnnpack_fp32.pte",
      "backend": "xnnpack",
      "precision": "fp32",
      "size_mb": 157.3
    }
  ],
  "note": "ExecuTorch .pte exports. See README.md for variant selection and run instructions."
}