config.json

{
  "architectures": [
    "HFUNetModel"
  ],
  "attention_resolutions": [
    16,
    32,
    64
  ],
  "channel_mult": [
    0.5,
    1,
    1,
    2,
    2,
    4,
    4
  ],
  "dropout": 0.0,
  "image_size": 512,
  "in_channels": 3,
  "model_channels": 256,
  "num_classes": null,
  "num_head_channels": 64,
  "num_heads": 4,
  "num_heads_upsample": -1,
  "num_res_blocks": 2,
  "out_channels": 6,
  "resblock_updown": true,
  "torch_dtype": "float32",
  "transformers_version": "4.22.2",
  "use_checkpoint": false,
  "use_fp16": true,
  "use_new_attention_order": false,
  "use_scale_shift_norm": true
}