unet/config.json

{
  "_class_name": "UNetUnconditionalModel",
  "_diffusers_version": "0.0.4",
  "attention_resolutions": [
    8,
    4,
    2
  ],
  "down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D"],
  "up_blocks": ["UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
  "down_block_input_channels": [224, 224, 448, 672],
  "down_block_output_channels": [224, 448, 672, 896],
  "context_dim": null,
  "conv_resample": true,
  "dims": 2,
  "dropout": 0,
  "image_size": 64,
  "in_channels": 3,
  "legacy": true,
  "n_embed": null,
  "num_classes": null,
  "num_head_channels": 32,
  "num_heads": -1,
  "num_heads_upsample": -1,
  "num_res_blocks": 2,
  "out_channels": 3,
  "resblock_updown": false,
  "transformer_depth": 1,
  "use_checkpoint": false,
  "use_fp16": false,
  "use_new_attention_order": false,
  "use_scale_shift_norm": false,
  "use_spatial_transformer": false,
  "ldm": true
}