DaKing16/ImageGenerationMachineLearning

1{
2  "_class_name": "UNet2DModel",
3  "_diffusers_version": "0.33.0.dev0",
4  "act_fn": "silu",
5  "add_attention": true,
6  "attention_head_dim": 8,
7  "attn_norm_num_groups": null,
8  "block_out_channels": [
9    128,
10    128,
11    256,
12    256,
13    512,
14    512
15  ],
16  "center_input_sample": false,
17  "class_embed_type": null,
18  "down_block_types": [
19    "DownBlock2D",
20    "DownBlock2D",
21    "DownBlock2D",
22    "DownBlock2D",
23    "AttnDownBlock2D",
24    "DownBlock2D"
25  ],
26  "downsample_padding": 1,
27  "downsample_type": "conv",
28  "dropout": 0.0,
29  "flip_sin_to_cos": true,
30  "freq_shift": 0,
31  "in_channels": 3,
32  "layers_per_block": 2,
33  "mid_block_scale_factor": 1,
34  "mid_block_type": "UNetMidBlock2D",
35  "norm_eps": 1e-05,
36  "norm_num_groups": 32,
37  "num_class_embeds": null,
38  "num_train_timesteps": null,
39  "out_channels": 3,
40  "resnet_time_scale_shift": "default",
41  "sample_size": 64,
42  "time_embedding_dim": null,
43  "time_embedding_type": "positional",
44  "up_block_types": [
45    "UpBlock2D",
46    "AttnUpBlock2D",
47    "UpBlock2D",
48    "UpBlock2D",
49    "UpBlock2D",
50    "UpBlock2D"
51  ],
52  "upsample_type": "conv"
53}
54