| { |
| "_class_name": "NCSNpp", |
| "_diffusers_version": "0.0.4", |
| "attention_type": "ddpm", |
| "attn_resolutions": [ |
| 16 |
| ], |
| "centered": false, |
| "ch_mult": [ |
| 1, |
| 2, |
| 2, |
| 2 |
| ], |
| "conditional": true, |
| "continuous": true, |
| "conv_size": 3, |
| "dropout": 0.0, |
| "embedding_type": "fourier", |
| "fir": true, |
| "fir_kernel": [ |
| 1, |
| 3, |
| 3, |
| 1 |
| ], |
| "fourier_scale": 16, |
| "image_size": 32, |
| "init_scale": 0.0, |
| "nf": 32, |
| "nonlinearity": "swish", |
| "normalization": "GroupNorm", |
| "num_channels": 3, |
| "num_res_blocks": 1, |
| "progressive": "output_skip", |
| "progressive_combine": "sum", |
| "progressive_input": "input_skip", |
| "resamp_with_conv": true, |
| "resblock_type": "biggan", |
| "scale_by_sigma": true, |
| "skip_rescale": true, |
| "block_channels": [32, 64, 64, 64], |
| "in_channels": 3, |
| "num_res_blocks": 1, |
| "out_channels": 3, |
| "time_embedding_type": "fourier", |
| "resnet_eps": 1e-6, |
| "mid_block_scale_factor": 1.41421356237, |
| "resnet_num_groups": null, |
| "down_blocks": [ |
| "UNetResSkipDownBlock2D", |
| "UNetResAttnSkipDownBlock2D", |
| "UNetResSkipDownBlock2D", |
| "UNetResSkipDownBlock2D" |
| ], |
| "up_blocks": [ |
| "UNetResSkipUpBlock2D", |
| "UNetResSkipUpBlock2D", |
| "UNetResAttnSkipUpBlock2D", |
| "UNetResSkipUpBlock2D" |
| ] |
| } |
|
|