File size: 2,660 Bytes
5b1aeb7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | {
"engine": "onnx",
"family": "decima",
"layout": "decima-late-interaction-v1",
"upstream": {
"repo": "amyrmahdy/decima-agent",
"revision": "86a07aab1c340fa5869bdb754e57d3851a1d288a",
"tag": "v2.1",
"checkpoint": "pytorch",
"code": {
"repo": "https://github.com/amyrmahdy/decima",
"tag": "v1.1.1",
"commit": "2df60942c68b0e1dfc87743462745bacdd936d8a"
},
"license": "Apache-2.0",
"base": "amyrmahdy/decima-base, on jhu-clsp/mmBERT-base (MIT)"
},
"contract": {
"inputs": {
"state_ids": {
"dtype": "int64",
"shape": [
"questions",
"state_len"
],
"note": "each question's state row, right-padded with special_tokens.pad"
},
"state_mask": {
"dtype": "int64",
"shape": [
"questions",
"state_len"
],
"note": "1 on tokens"
},
"option_ids": {
"dtype": "int64",
"shape": [
"options",
"option_len"
],
"note": "every option row, questions in order, right-padded"
},
"option_mask": {
"dtype": "int64",
"shape": [
"options",
"option_len"
],
"note": "1 on tokens"
},
"option_state": {
"dtype": "int64",
"shape": [
"options"
],
"note": "the question (state row) of each option"
}
},
"outputs": {
"scores": {
"dtype": "float32",
"shape": [
"options"
],
"note": "raw option scores; choice and noul: softmax(scores / temperature)"
},
"ordinal_g": {
"dtype": "float32",
"shape": [
"options"
],
"note": "ord_g(z) per option"
},
"ordinal_gap": {
"dtype": "float32",
"shape": [
"options"
],
"note": "ord_gap(z) per option"
}
}
},
"state_prefix": "",
"option_prefix": "",
"question_in_state": true,
"question_in_options": true,
"max_state_tokens": 2048,
"max_option_tokens": 64,
"special_tokens": {
"cls": 2,
"sep": 1,
"pad": 0
},
"noul": {
"options": [
"yes",
"no"
],
"criteria": "\nTrue if: {true}\nFalse if: {false}",
"missing": "\u2014"
},
"limits": {
"questions": [
1,
256
],
"choices": [
2,
255
],
"levels": [
2,
10
]
},
"temperature": 0.6081955235647175,
"option_logits": {
"choice": "labels in the caller's order",
"score": "levels in order",
"noul": "[yes, no]"
},
"opset": 20
}
|