{
  "architectures": [
    "CXRMateEDModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_cxrmate_ed.CXRMateEDConfig",
    "AutoModel": "modelling_cxrmate_ed.CXRMateEDModel"
  },
  "decoder": {
    "add_time_deltas": true,
    "hidden_size": 768,
    "history": 0,
    "include_time_delta": true,
    "index_value_encoder_intermediate_size": 2048,
    "intermediate_size": 3072,
    "is_decoder": true,
    "model_type": "llama",
    "num_attention_heads": 12,
    "num_hidden_layers": 6,
    "num_key_value_heads": 12,
    "pad_token_id": 4,
    "prompt_report_sections_filter": [
      "indication",
      "history"
    ],
    "tables_filter": [
      "mimic_cxr_sectioned",
      "triage",
      "medrecon"
    ],
    "time_delta_monotonic_inversion": true,
    "vocab_size": 30000
  },
  "encoder": {
    "_name_or_path": "aehrc/uniformer_base_tl_384",
    "architectures": [
      "UniFormerModel"
    ],
    "auto_map": {
      "AutoConfig": "aehrc/uniformer_base_tl_384--configuration_uniformer.UniFormerWithProjectionHeadConfig",
      "AutoModel": "aehrc/uniformer_base_tl_384--modelling_uniformer.UniFormerModel"
    },
    "init_value": 1e-06,
    "layer_scale": false,
    "model_type": "uniformer",
    "projection_size": 768,
    "torch_dtype": "float32"
  },
  "is_encoder_decoder": false,
  "model_type": "cxrmate-ed",
  "tie_word_embeddings": false,
  "torch_dtype": "float32",
  "transformers_version": "4.39.3"
}