onnx_config.json

{
  "format": "onnx",
  "model_type": "token_classification",
  "inputs": {
    "input_ids": {
      "dtype": "int64",
      "shape": [
        "batch",
        "sequence"
      ]
    },
    "attention_mask": {
      "dtype": "int64",
      "shape": [
        "batch",
        "sequence"
      ]
    }
  },
  "outputs": {
    "logits": {
      "dtype": "float32",
      "shape": [
        "batch",
        "sequence",
        2
      ]
    }
  }
}