| { |
| "engine": "onnx", |
| "family": "decima", |
| "layout": "decima-late-interaction-v1", |
| "upstream": { |
| "repo": "amyrmahdy/decima-base", |
| "revision": "2468005d5e48e95eb74072c32a6d9df164578071", |
| "tag": "v2.0", |
| "checkpoint": "pytorch", |
| "code": { |
| "repo": "https://github.com/amyrmahdy/decima", |
| "tag": "v1.1.1", |
| "commit": "2df60942c68b0e1dfc87743462745bacdd936d8a" |
| }, |
| "license": "Apache-2.0", |
| "base": "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": 512, |
| "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.7269418400253115, |
| "option_logits": { |
| "choice": "labels in the caller's order", |
| "score": "levels in order", |
| "noul": "[yes, no]" |
| }, |
| "opset": 20 |
| } |
|
|
