upscale_unet/config.json

{
  "_class_name": "GLIDESuperResUNetModel",
  "_diffusers_version": "0.0.3",
  "attention_resolutions": [
    8,
    16,
    32
  ],
  "channel_mult": [
    1,
    1,
    2,
    2,
    4,
    4
  ],
  "conv_resample": true,
  "dims": 2,
  "dropout": 0.1,
  "in_channels": 6,
  "model_channels": 192,
  "num_head_channels": 64,
  "num_heads": 1,
  "num_heads_upsample": 1,
  "num_res_blocks": 2,
  "out_channels": 6,
  "resblock_updown": true,
  "resolution": 256,
  "use_checkpoint": false,
  "use_fp16": false,
  "use_scale_shift_norm": true
}