Safetensors
rt_detr_v2
docling-layout-heron / bundle.json
tomsanbear's picture
Add validated Docling Heron bundle
63d9148 verified
Raw
History Blame Contribute Delete
2.23 kB
{
"architecture_id": "layout.heron-rtdetr",
"bundle_id": "layout/heron-rtdetr",
"license": "Apache-2.0",
"parity_source": {
"files": [
"model.onnx",
"config.json",
"preprocessor_config.json"
],
"repo": "docling-project/docling-layout-heron-onnx",
"revision": "40bde044036bb181c130ddf6c51792187268748f"
},
"provenance": {
"conversion": "native safetensors copied byte-for-byte; metadata generated",
"legacy_predictor_boundary": "images u8 + orig_target_sizes i64 to labels/boxes/scores",
"native_input": "RTDetrImageProcessor pixel_values f32, RGB, rescaled by 1/255, not normalized",
"orig_target_sizes_order": "width,height when reproducing the legacy fused tail",
"runtime_onnx": "false",
"source_model_class": "RTDetrV2ForObjectDetection"
},
"required_files": [
{
"path": "model.safetensors",
"role": "weights"
},
{
"path": "config.json",
"role": "config"
},
{
"path": "preprocessor_config.json",
"role": "preprocessor"
},
{
"path": "labels.json",
"role": "label-map"
},
{
"path": "README.md",
"role": "source-card"
},
{
"path": "tensor_map.json",
"role": "tensor-map"
},
{
"path": "validation.json",
"role": "validation"
}
],
"schema_version": 1,
"source": {
"files": [
"README.md",
"config.json",
"model.safetensors",
"preprocessor_config.json"
],
"repo": "docling-project/docling-layout-heron",
"revision": "8f39ad3c0b4c58e9c2d2c84a38465abf757272d8"
},
"tensor_contract": {
"inputs": [
{
"dtype": "f32",
"name": "pixel_values",
"shape": [
"batch",
3,
640,
640
]
}
],
"outputs": [
{
"dtype": "f32",
"name": "logits",
"shape": [
"batch",
300,
17
]
},
{
"dtype": "f32",
"name": "pred_boxes",
"shape": [
"batch",
300,
4
]
}
]
},
"validated_runtime": [
{
"device": "cpu",
"dtype": "f32"
}
]
}