| { | |
| "_class_name": "NCSNpp", | |
| "_diffusers_version": "0.0.4", | |
| "attention_type": "ddpm", | |
| "attn_resolutions": [ | |
| 16 | |
| ], | |
| "centered": false, | |
| "ch_mult": [ | |
| 1, | |
| 2, | |
| 4, | |
| 8, | |
| 16, | |
| 32, | |
| 32, | |
| 32 | |
| ], | |
| "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": 1024, | |
| "init_scale": 0.0, | |
| "nf": 16, | |
| "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": [16, 32, 64, 128, 256, 512, 512, 512], | |
| "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", | |
| "UNetResSkipDownBlock2D", | |
| "UNetResSkipDownBlock2D", | |
| "UNetResSkipDownBlock2D", | |
| "UNetResSkipDownBlock2D", | |
| "UNetResSkipDownBlock2D", | |
| "UNetResAttnSkipDownBlock2D", | |
| "UNetResSkipDownBlock2D" | |
| ], | |
| "up_blocks": [ | |
| "UNetResSkipUpBlock2D", | |
| "UNetResAttnSkipUpBlock2D", | |
| "UNetResSkipUpBlock2D", | |
| "UNetResSkipUpBlock2D", | |
| "UNetResSkipUpBlock2D", | |
| "UNetResSkipUpBlock2D", | |
| "UNetResSkipUpBlock2D", | |
| "UNetResSkipUpBlock2D" | |
| ], | |
| "center_input_sample": true, | |
| "num_head_channels": null | |
| } | |