dyllanesl commited on
Commit
aebbfcc
·
verified ·
1 Parent(s): 439b428

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -21
config.json CHANGED
@@ -2,30 +2,18 @@
2
  "_class_name": "UNet2DModel",
3
  "_diffusers_version": "0.29.2",
4
  "_name_or_path": "google/ddpm-cifar10-32",
5
- "act_fn": "silu",
6
- "add_attention": true,
7
- "attention_head_dim": null,
8
- "attn_norm_num_groups": null,
9
  "block_out_channels": [128, 256, 256, 256],
10
- "center_input_sample": false,
11
- "class_embed_type": null,
12
  "down_block_types": ["DownBlock2D", "AttnDownBlock2D", "DownBlock2D", "DownBlock2D"],
13
- "downsample_padding": 0,
14
- "downsample_type": "conv",
15
  "dropout": 0.0,
16
- "flip_sin_to_cos": false,
17
- "freq_shift": 1,
18
- "in_channels": 3,
19
- "layers_per_block": 2,
20
- "mid_block_scale_factor": 1,
21
- "norm_eps": 1e-06,
22
  "norm_num_groups": 32,
23
- "num_class_embeds": null,
24
- "num_train_timesteps": null,
25
- "out_channels": 3,
26
- "resnet_time_scale_shift": "default",
27
- "sample_size": 32,
28
  "time_embedding_type": "positional",
29
- "up_block_types": ["UpBlock2D", "UpBlock2D", "AttnUpBlock2D", "UpBlock2D"],
30
- "upsample_type": "conv"
31
  }
 
2
  "_class_name": "UNet2DModel",
3
  "_diffusers_version": "0.29.2",
4
  "_name_or_path": "google/ddpm-cifar10-32",
5
+ "sample_size": 32,
6
+ "in_channels": 3,
7
+ "out_channels": 3,
8
+ "layers_per_block": 2,
9
  "block_out_channels": [128, 256, 256, 256],
 
 
10
  "down_block_types": ["DownBlock2D", "AttnDownBlock2D", "DownBlock2D", "DownBlock2D"],
11
+ "up_block_types": ["UpBlock2D", "UpBlock2D", "AttnUpBlock2D", "UpBlock2D"],
 
12
  "dropout": 0.0,
13
+ "act_fn": "silu",
 
 
 
 
 
14
  "norm_num_groups": 32,
15
+ "norm_eps": 1e-06,
16
+ "mid_block_scale_factor": 1,
 
 
 
17
  "time_embedding_type": "positional",
18
+ "add_attention": true
 
19
  }