ffhq_ncsnpp / config.json
patrickvonplaten's picture
Update config.json
fc7feab
{
"_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
}