rubert-base-pii-ner-tensorrt / export_metadata.json
lockR's picture
Add validated ruBERT PII inference export, model card and benchmarks
73be581 verified
Raw History Blame Contribute Delete
5.29 kB
{
"model_id": "redmadrobot-rnd/rubert-base-pii-ner",
"model_revision": "c802e8cd26f85d1cf920973ea6f83965a0618d63",
"model_url": "https://huggingface.co/redmadrobot-rnd/rubert-base-pii-ner/tree/c802e8cd26f85d1cf920973ea6f83965a0618d63",
"source_directory": "artifacts/source",
"created_at_utc": "2026-09-22T19:49:34.071363+00:00",
"versions": {
"torch": "2.8.0",
"transformers": "4.57.1",
"onnx": "1.19.1",
"onnxruntime-gpu": "1.23.2",
"huggingface-hub": "0.36.2"
},
"export": {
"device": "cpu",
"opset": 17,
"dynamo": false,
"attention_implementation": "eager",
"input_dtype": "int64",
"output_dtype": "float32",
"dynamic_axes": {
"input_ids": {
"0": "batch",
"1": "sequence"
},
"attention_mask": {
"0": "batch",
"1": "sequence"
},
"token_type_ids": {
"0": "batch",
"1": "sequence"
},
"logits": {
"0": "batch",
"1": "sequence"
}
},
"elapsed_seconds": 11.209364499998628,
"created_at_utc": "2026-09-22T19:49:34.071363+00:00",
"versions": {
"torch": "2.8.0",
"transformers": "4.57.1",
"onnx": "1.19.1",
"onnxruntime-gpu": "1.23.2",
"huggingface-hub": "0.36.2"
}
},
"optimizer": {
"model_type": "bert",
"num_heads": 12,
"hidden_size": 768,
"opt_level": 0,
"use_gpu": true,
"keep_io_types": true,
"fusion_statistics": {
"EmbedLayerNormalization": 1,
"Attention": 12,
"MultiHeadAttention": 0,
"Gelu": 0,
"FastGelu": 0,
"BiasGelu": 12,
"GemmFastGelu": 0,
"LayerNormalization": 0,
"SimplifiedLayerNormalization": 0,
"SkipLayerNormalization": 24,
"SkipSimplifiedLayerNormalization": 0,
"RotaryEmbedding": 0,
"QOrderedAttention": 0,
"QOrderedGelu": 0,
"QOrderedLayerNormalization": 0,
"QOrderedMatMul": 0
},
"elapsed_seconds": 22.08802573200228,
"created_at_utc": "2026-09-22T19:49:34.071363+00:00",
"versions": {
"torch": "2.8.0",
"transformers": "4.57.1",
"onnx": "1.19.1",
"onnxruntime-gpu": "1.23.2",
"huggingface-hub": "0.36.2"
}
},
"artifacts": {
"onnx_fp32": {
"path": "artifacts/model.onnx",
"bytes": 711206326,
"sha256": "fa8775d59ef345d7934fb7a01671fbe24c5c098c0e44dcc397a67af6c5a4805b",
"nodes": 879,
"operators": {
"ai.onnx::Add": 123,
"ai.onnx::Cast": 1,
"ai.onnx::Concat": 48,
"ai.onnx::Constant": 274,
"ai.onnx::Div": 24,
"ai.onnx::Erf": 12,
"ai.onnx::Gather": 40,
"ai.onnx::LayerNormalization": 25,
"ai.onnx::MatMul": 97,
"ai.onnx::Mul": 25,
"ai.onnx::Reshape": 48,
"ai.onnx::Shape": 37,
"ai.onnx::Slice": 1,
"ai.onnx::Softmax": 12,
"ai.onnx::Sub": 1,
"ai.onnx::Transpose": 48,
"ai.onnx::Unsqueeze": 63
},
"inputs": [
"input_ids",
"attention_mask",
"token_type_ids"
],
"outputs": [
"logits"
]
},
"ort_fp32": {
"path": "artifacts/model.optimized.fp32.onnx",
"bytes": 711034332,
"sha256": "eebed6562883ca6c59204da661ddb3edb1b334e91737f35d79ac74c75d371315",
"nodes": 90,
"operators": {
"ai.onnx::Add": 1,
"ai.onnx::Cast": 3,
"ai.onnx::MatMul": 37,
"com.microsoft::Attention": 12,
"com.microsoft::BiasGelu": 12,
"com.microsoft::EmbedLayerNormalization": 1,
"com.microsoft::SkipLayerNormalization": 24
},
"inputs": [
"input_ids",
"attention_mask",
"token_type_ids"
],
"outputs": [
"logits"
]
},
"ort_fp16": {
"path": "artifacts/model.optimized.fp16.onnx",
"bytes": 355541615,
"sha256": "039aba6be0c64f50f37c64ddc6c6d4e2fd91cbdcf5c9291056ec6bc96c70a501",
"nodes": 91,
"operators": {
"ai.onnx::Add": 1,
"ai.onnx::Cast": 4,
"ai.onnx::MatMul": 37,
"com.microsoft::Attention": 12,
"com.microsoft::BiasGelu": 12,
"com.microsoft::EmbedLayerNormalization": 1,
"com.microsoft::SkipLayerNormalization": 24
},
"inputs": [
"input_ids",
"attention_mask",
"token_type_ids"
],
"outputs": [
"logits"
]
}
},
"source_files": {
"LICENSE": {
"bytes": 11358,
"sha256": "cfc7749b96f63bd31c3c42b5c471bf756814053e847c10f3eb003417bc523d30"
},
"README.md": {
"bytes": 6128,
"sha256": "2bfb4bff70ebcf0ece0cd83c96f44e5d179c7fd5dbe20c5fdcaf2592845900c8"
},
"config.json": {
"bytes": 2831,
"sha256": "f4eeba88d1503a56d629195c9307d42c1acf4ce53ab88fdaecfc754fd0745cf9"
},
"model.safetensors": {
"bytes": 711022548,
"sha256": "2b9510cd99101c05800c3b4bcaeaebb26d285aa3152e49bea09dc5c2a38d4b16"
},
"tokenizer.json": {
"bytes": 3714701,
"sha256": "f61eee9d0dc18f14c195321a37eadf9e5aeb3a323bc278dec689adeb89dec1b9"
},
"tokenizer_config.json": {
"bytes": 322,
"sha256": "ffe381e31042ba94cb077bf7d5b9cee944dbd0c2a85079b095bea7e5a5a6058b"
}
}
}