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