{
  "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
}