{ "_class_name": "UNet2DModel", "_diffusers_version": "0.0.4", "act_fn": "silu", "attention_head_dim": null, "block_channels": [ 224, 448, 672, 896 ], "block_out_channels": [ 16, 32, 64, 128, 256, 512, 512, 512 ], "center_input_sample": true, "down_blocks": [ "SkipDownBlock2D", "SkipDownBlock2D", "SkipDownBlock2D", "SkipDownBlock2D", "SkipDownBlock2D", "SkipDownBlock2D", "AttnSkipDownBlock2D", "SkipDownBlock2D" ], "downsample_padding": 1, "downscale_freq_shift": 0, "flip_sin_to_cos": true, "freq_shift": 0, "image_size": null, "in_channels": 3, "layers_per_block": 1, "mid_block_scale_factor": 1.41421356237, "norm_eps": 1e-06, "norm_num_groups": null, "num_head_channels": 32, "num_res_blocks": null, "out_channels": 3, "resnet_act_fn": "silu", "resnet_eps": 1e-05, "resnet_num_groups": 32, "sample_size": 1024, "time_embedding_type": "fourier", "up_blocks": [ "SkipUpBlock2D", "AttnSkipUpBlock2D", "SkipUpBlock2D", "SkipUpBlock2D", "SkipUpBlock2D", "SkipUpBlock2D", "SkipUpBlock2D", "SkipUpBlock2D" ] }