Add the ONNX bundle bibr's ML_RUNTIME=onnx path loads
Browse files- onnx/bibr_onnx.json +80 -0
- onnx/model.onnx +3 -0
- onnx/tokenizer.json +0 -0
onnx/bibr_onnx.json
ADDED
|
@@ -0,0 +1,80 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architecture": "SectionMiniLMModel",
|
| 3 |
+
"exporter": "torchscript",
|
| 4 |
+
"files": [
|
| 5 |
+
"model.onnx",
|
| 6 |
+
"tokenizer.json"
|
| 7 |
+
],
|
| 8 |
+
"inputs": [
|
| 9 |
+
{
|
| 10 |
+
"dtype": "int64",
|
| 11 |
+
"name": "input_ids",
|
| 12 |
+
"shape": [
|
| 13 |
+
"batch",
|
| 14 |
+
"sequence"
|
| 15 |
+
]
|
| 16 |
+
},
|
| 17 |
+
{
|
| 18 |
+
"dtype": "int64",
|
| 19 |
+
"name": "attention_mask",
|
| 20 |
+
"shape": [
|
| 21 |
+
"batch",
|
| 22 |
+
"sequence"
|
| 23 |
+
]
|
| 24 |
+
}
|
| 25 |
+
],
|
| 26 |
+
"label_classes": [
|
| 27 |
+
"abstract",
|
| 28 |
+
"acknowledgment",
|
| 29 |
+
"author_contributions",
|
| 30 |
+
"coi",
|
| 31 |
+
"discussion",
|
| 32 |
+
"endnote",
|
| 33 |
+
"ethics",
|
| 34 |
+
"funding",
|
| 35 |
+
"intro",
|
| 36 |
+
"keywords",
|
| 37 |
+
"method",
|
| 38 |
+
"open_data",
|
| 39 |
+
"references",
|
| 40 |
+
"results",
|
| 41 |
+
"title",
|
| 42 |
+
"unknown"
|
| 43 |
+
],
|
| 44 |
+
"max_length": 256,
|
| 45 |
+
"model": "section_classifier",
|
| 46 |
+
"model_file": "model.onnx",
|
| 47 |
+
"opset": 17,
|
| 48 |
+
"outputs": [
|
| 49 |
+
{
|
| 50 |
+
"name": "type_logits",
|
| 51 |
+
"shape": [
|
| 52 |
+
"batch",
|
| 53 |
+
16
|
| 54 |
+
]
|
| 55 |
+
},
|
| 56 |
+
{
|
| 57 |
+
"name": "top_level_logits",
|
| 58 |
+
"shape": [
|
| 59 |
+
"batch"
|
| 60 |
+
]
|
| 61 |
+
}
|
| 62 |
+
],
|
| 63 |
+
"pad_token": "[PAD]",
|
| 64 |
+
"pad_token_id": 0,
|
| 65 |
+
"schema_version": 1,
|
| 66 |
+
"source": {
|
| 67 |
+
"exported_at": "2026-09-02T20:32:41+00:00",
|
| 68 |
+
"numpy": "2.3.5",
|
| 69 |
+
"onnx": "1.22.0",
|
| 70 |
+
"onnxruntime": "1.27.0",
|
| 71 |
+
"platform": "Linux-7.2.2-1-cachyos-x86_64-with-glibc2.44",
|
| 72 |
+
"python": "3.12.14",
|
| 73 |
+
"repo_id": "scienceverse/bibr-section-classifier",
|
| 74 |
+
"revision": "969112772a8940d46c5d6b19feee45eeebfd26f9",
|
| 75 |
+
"torch": "2.11.0+cu130",
|
| 76 |
+
"transformers": "5.14.1"
|
| 77 |
+
},
|
| 78 |
+
"template_version": 3,
|
| 79 |
+
"tokenizer_file": "tokenizer.json"
|
| 80 |
+
}
|
onnx/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:474de26402d09ddf8db15308b9ebe2e6f8251856d895fccdbf6d18ec5e0ea707
|
| 3 |
+
size 90427110
|
onnx/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|