{
  "architectures": [
    "CondUNetForSemanticSegmentation"
  ],
  "attn_start": 0,
  "auto_map": {
    "AutoConfig": "configuration_cond_unet.CondUNetConfig",
    "AutoImageProcessor": "image_processing_cond_unet.CondUNetImageProcessor",
    "AutoModelForImageSegmentation": "modeling_cond_unet.CondUNetForSemanticSegmentation"
  },
  "custom_pipelines": {
    "image-segmentation": {
      "impl": "pipeline.CondUNetImageSegmentationPipeline",
      "pt": [
        "AutoModelForImageSegmentation"
      ],
      "type": "image"
    }
  },
  "depth": 5,
  "dtype": "float32",
  "dwt_bands": [
    "LL",
    "LH",
    "HL",
    "HH"
  ],
  "emb_dim": 768,
  "id2label": {
    "0": "foreground"
  },
  "image_size": 512,
  "in_channels": 3,
  "keep_aspect_ratio": true,
  "label2id": {
    "foreground": 0
  },
  "model_type": "cond_unet",
  "n_heads": 8,
  "n_organs": 10,
  "patch_size": 8,
  "self_normalize": true,
  "shape_res": 32,
  "size": 16,
  "transformers_version": "5.16.1",
  "unknown_organ_id": -1,
  "use_attn": true,
  "use_dwt": false,
  "use_shape": false,
  "wavelet": "haar"
}