Safetensors
rt_detr_v2
docling-layout-heron / validation.json
tomsanbear's picture
Add validated Docling Heron bundle
63d9148 verified
Raw
History Blame Contribute Delete
4.02 kB
{
"conversion_tool": "tools/models/candle/convert_heron.py@schema-v1",
"dtype_device_results": [
{
"device": "cpu",
"dtype": "f32",
"note": "native safetensors metadata and raw RT-DETR shape contract smoke",
"status": "passed"
},
{
"device": "cuda",
"dtype": "f16",
"note": "accelerator validation deferred until native Heron forward exists",
"status": "unvalidated"
},
{
"device": "metal",
"dtype": "f16",
"note": "accelerator validation deferred until native Heron forward exists",
"status": "unvalidated"
}
],
"generated_files": [
{
"path": "model.safetensors",
"sha256": "00333a43451945aaf89db8ca9c0a17e75d1537c17db60fdb91aa95f4c7929e0c",
"size_bytes": 171658996
},
{
"path": "config.json",
"sha256": "fdea30805ce2f5666b147fca941dcdd27ad468e27d6ed21902207d3da056a97d",
"size_bytes": 3268
},
{
"path": "preprocessor_config.json",
"sha256": "cd38cd59999e7a95d68e487fbe5132df3d4e5c32a0836add57e6126ba0c4eaf1",
"size_bytes": 444
},
{
"path": "labels.json",
"sha256": "000e33b56ffaa828244777692f7de18bb496e3112710cf303c1b6b17bbdea7db",
"size_bytes": 1361
},
{
"path": "README.md",
"sha256": "175700839bc7808eac6af1d0c23e4f483606ab2276fe01122f4093e61a1a65b6",
"size_bytes": 3219
},
{
"path": "tensor_map.json",
"sha256": "7077af1303c8fbcb584a9faf4584a4d364b099d0fc0bacdc24ca5448fb12bbe0",
"size_bytes": 231669
},
{
"path": "bundle.json",
"sha256": "729bb0e9f8d239d61d4b41debf6ac2568a14b7142f88edb791346ed5e741b6f9",
"size_bytes": 2232
}
],
"parity_fixture_ids": [
"heron-onnx-synthetic-gray-layout",
"heron-onnx-omnidocbench-1653x2339-layout"
],
"schema_version": 1,
"source_files": [
{
"path": "README.md",
"sha256": "175700839bc7808eac6af1d0c23e4f483606ab2276fe01122f4093e61a1a65b6",
"size_bytes": 3219
},
{
"path": "config.json",
"sha256": "fdea30805ce2f5666b147fca941dcdd27ad468e27d6ed21902207d3da056a97d",
"size_bytes": 3268
},
{
"path": "model.safetensors",
"sha256": "00333a43451945aaf89db8ca9c0a17e75d1537c17db60fdb91aa95f4c7929e0c",
"size_bytes": 171658996
},
{
"path": "preprocessor_config.json",
"sha256": "cd38cd59999e7a95d68e487fbe5132df3d4e5c32a0836add57e6126ba0c4eaf1",
"size_bytes": 444
}
],
"tested_io": [
{
"id": "heron-rtdetr-native-raw-shape-smoke",
"inputs": [
{
"dtype": "f32",
"name": "pixel_values",
"shape": [
1,
3,
640,
640
]
}
],
"outputs": [
{
"dtype": "f32",
"name": "logits",
"shape": [
1,
300,
17
]
},
{
"dtype": "f32",
"name": "pred_boxes",
"shape": [
1,
300,
4
]
}
]
},
{
"id": "heron-legacy-predictor-boundary-parity-contract",
"inputs": [
{
"dtype": "u8",
"name": "images",
"shape": [
1,
3,
640,
640
]
},
{
"dtype": "i64",
"name": "orig_target_sizes",
"shape": [
1,
2
]
}
],
"outputs": [
{
"dtype": "i64",
"name": "labels",
"shape": [
1,
300
]
},
{
"dtype": "f32",
"name": "boxes",
"shape": [
1,
300,
4
]
},
{
"dtype": "f32",
"name": "scores",
"shape": [
1,
300
]
}
]
}
]
}