config.json

{
  "_class_name": "UNetModel",
  "attn_resolutions": [
    16
  ],
  "ch": 32,
  "ch_mult": [
    1,
    2
  ],
  "dropout": 0.0,
  "in_channels": 3,
  "num_res_blocks": 2,
  "out_ch": 3,
  "resamp_with_conv": true,
  "resolution": 32
}