{
  "trunk_model": {
    "pretrained_model_name_or_path": "/checkpoint/sarah/repos/srch-search-api-nemotron/python/citras/outputs/26-09-02/14_10_22_110509/checkpoints/step-3750/hf_model_merged",
    "revision": null,
    "truncate_dim": null,
    "hf_kwargs": {
      "attn_implementation": "sdpa",
      "torch_dtype": "auto"
    },
    "lora_config": null,
    "input_mapper": {
      "_target_": "citras.data.ContextualInputMapper",
      "chunk_regex": "[^.!?\\n]+(?:[.!?]+|\\n+|$)",
      "chunk_splitter": "segtok",
      "max_chunks": null,
      "query_prefix": null,
      "document_prefix": null,
      "boundary_marker": "<|endoftext|>"
    }
  },
  "heads": {
    "contextual": {
      "_target_": "citras.model.ContextualHead",
      "input_dim": 2560,
      "embedding_dim": null,
      "chunk_pooling_strategy": "mean_pooling",
      "quantizer": null
    }
  }
}