{
  "source_model": "cua-ai/cua-s1-forms",
  "format": "onnx",
  "onnx_file": "cua-s1-forms.onnx",
  "inputs": {
    "context_ids": "int64 [batch, context_length]",
    "context_mask": "bool [batch, context_length]",
    "option_ids": "int64 [batch, options, option_length]",
    "option_token_mask": "bool [batch, options, option_length]",
    "option_mask": "bool [batch, options]"
  },
  "output": {
    "logits": "float32 [batch, options]"
  },
  "byte_encoding": "UTF-8 byte + 1; 0 is padding",
  "config": {
    "encoder": "tinyx",
    "hf_model": "Qwen/Qwen2.5-0.5B",
    "width": 128,
    "rank": 128,
    "layers": 2,
    "heads": 4,
    "context_tokens": 224,
    "option_tokens": 96
  }
}