BackTo2014 commited on
Commit
4884249
1 Parent(s): baf3d19

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +27 -28
config.json CHANGED
@@ -1,33 +1,32 @@
1
  {
2
- "in_channels": 3,
3
- "out_channels": 3,
4
- "down_block_types": [
5
- "DownBlock2D",
6
- "AttnDownBlock2D",
7
- "DownBlock2D"
 
8
  ],
9
- "up_block_types": [
10
- "UpBlock2D",
11
- "AttnUpBlock2D",
12
- "UpBlock2D"
 
13
  ],
14
- "block_out_channels": [
15
- 128,
16
- 256,
17
- 384,
18
- 512
19
  ],
20
- "layers_per_block": 2,
21
- "norm_num_groups": 32,
22
- "cross_attention_dim": null,
23
- "attention_head_dim": 8,
24
- "resnet_time_scale_shift": "default",
25
- "class_embed_type": null,
26
- "num_class_embeds": null,
27
- "use_linear_projection": false,
28
- "only_cross_attention": false,
29
- "mid_block_scale_factor": 1,
30
- "out_bias": true,
31
- "flip_sin_to_cos": true,
32
- "freq_shift": 0
33
  }
 
1
  {
2
+ "in_channels": 3,
3
+ "out_channels": 3,
4
+ "down_block_types": [
5
+ "ResBlock", "ResBlock", "DownSample",
6
+ "ResBlock", "AttnResBlock", "ResBlock", "DownSample",
7
+ "ResBlock", "ResBlock", "DownSample",
8
+ "ResBlock", "ResBlock"
9
  ],
10
+ "up_block_types": [
11
+ "ResBlock", "ResBlock",
12
+ "ResBlock", "ResBlock", "DownSample",
13
+ "ResBlock", "AttnResBlock", "ResBlock", "DownSample",
14
+ "ResBlock", "ResBlock", "DownSample"
15
  ],
16
+ "block_out_channels": [
17
+ 128, 256, 256, 256
 
 
 
18
  ],
19
+ "layers_per_block": 2,
20
+ "norm_num_groups": 32,
21
+ "cross_attention_dim": null,
22
+ "attention_head_dim": 8,
23
+ "resnet_time_scale_shift": "default",
24
+ "class_embed_type": null,
25
+ "num_class_embeds": null,
26
+ "use_linear_projection": false,
27
+ "only_cross_attention": false,
28
+ "mid_block_scale_factor": 1,
29
+ "out_bias": true,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0
32
  }