| { | |
| "_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 | |
| } | |