config.json

{
  "_class_name": "EDMUnet2D",
  "_diffusers_version": "0.30.3",
  "attn_resolutions": [],
  "block_kwargs": {},
  "concat_balance": 0.5,
  "conditional_inputs": [
    [
      "float",
      64,
      0.2
    ],
    [
      "float",
      64,
      0.2
    ],
    [
      "float",
      64,
      0.2
    ],
    [
      "float",
      64,
      0.2
    ],
    [
      "float",
      64,
      0.2
    ]
  ],
  "disable_out_gain": false,
  "emb_channels": null,
  "encode_only": false,
  "fourier_scale": "pos",
  "image_size": 16,
  "in_channels": 11,
  "layers_per_block": 2,
  "logvar_channels": 128,
  "midblock_attention": false,
  "model_channel_mults": [
    1
  ],
  "model_channels": 128,
  "n_logvar": 1,
  "noise_emb_dims": null,
  "out_channels": 6
}