config.json

{
  "architectures": [
    "JoliaModel"
  ],
  "atlas_config": {
    "modalities": {
      "abdomen_ct": {
        "image_size": [
          192,
          192,
          192
        ],
        "in_chans": 11,
        "local2global": [
          4,
          4,
          4
        ],
        "merge_ratio": [
          4,
          4,
          4
        ],
        "num_features": 96,
        "patch_size": [
          6,
          6,
          6
        ]
      }
    },
    "model": {
      "atlas_args": {
        "attn_drop": 0.0,
        "dim": 192,
        "drop": 0.0,
        "drop_path": 0.3,
        "mlp_ratio": 4.0,
        "num_heads": 6,
        "qkv_bias": true
      },
      "stages": [
        2,
        2,
        8
      ]
    }
  },
  "auto_map": {
    "AutoConfig": "configuration_jolia.JoliaConfig",
    "AutoModel": "modeling_jolia.JoliaModel"
  },
  "dtype": "float32",
  "embed_dim": 576,
  "model_type": "jolia",
  "multiscale_feats": true,
  "num_heads": 1,
  "num_organs": 200,
  "num_scales": 3,
  "organ_slot_names": [
    "abdominal_cavity",
    "abdominal_wall",
    "adrenal_glands",
    "anus",
    "aorta",
    "appendix",
    "atrial_appendage",
    "biliary",
    "bladder",
    "bone",
    "brachiocephalic_trunk",
    "brachiocephalic_vein",
    "brain",
    "breast",
    "chest",
    "chest_soft_tissue",
    "chest_soft_tissue_solid_mass",
    "chest_soft_tissues",
    "chest_wall",
    "clavicles",
    "colon",
    "common_carotid_artery",
    "device",
    "diaphragm",
    "duodenum",
    "epididymis",
    "esophagus",
    "fallopian_tube",
    "femurs",
    "gallbladder",
    "gastrointestinal_tract",
    "gonadal_vein",
    "heart",
    "hepatic_artery",
    "hepatic_vein",
    "hernia",
    "hip",
    "iliac_artery",
    "iliac_vein",
    "inferior_mesenteric_artery",
    "inferior_mesenteric_vein",
    "inferior_vena_cava",
    "inguinal_hernia",
    "joint",
    "kidneys",
    "large_bowel",
    "liver",
    "lungs",
    "lymph_nodes",
    "mediastinum",
    "mesenteric",
    "muscle",
    "orbit",
    "ovarian_artery",
    "ovarian_vein",
    "ovaries",
    "pancreas",
    "pelvis",
    "penis",
    "pericardium",
    "peritoneum",
    "pleura",
    "portal_artery",
    "portal_vein",
    "postsurgical_changes",
    "prostate",
    "pulmonary_artery",
    "pulmonary_vein",
    "rectum",
    "renal_artery",
    "renal_vein",
    "retroperitoneum",
    "ribs",
    "sacrum",
    "scapulae",
    "scrotum",
    "seminal_vesicle",
    "sinus",
    "skin",
    "skull",
    "small_bowel",
    "soft_tissues",
    "spinal_cord",
    "spine",
    "spleen",
    "splenic_artery",
    "splenic_vein",
    "sternum",
    "stomach",
    "subclavian_artery",
    "subclavian_vein",
    "superior_mesenteric_artery",
    "superior_mesenteric_vein",
    "superior_vena_cava",
    "testis",
    "thyroid",
    "total_organs",
    "trachea",
    "ureter",
    "urinary",
    "uterus",
    "vagina"
  ],
  "patch_embed_dim": 192,
  "text_context_length": 512,
  "text_embed_dim": 4096,
  "text_encoder_id": "Qwen/Qwen3-Embedding-8B",
  "transformers_version": "5.3.0",
  "use_cls_residual": false
}