ehristoforu commited on
Commit
b747994
1 Parent(s): 19d76c5

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +10 -7
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "act_fn": "silu",
5
  "addition_embed_type": "text_time",
6
  "addition_embed_type_num_heads": 64,
@@ -10,7 +10,6 @@
10
  10,
11
  20
12
  ],
13
- "attention_type": "default",
14
  "block_out_channels": [
15
  320,
16
  640,
@@ -29,7 +28,6 @@
29
  "CrossAttnDownBlock2D"
30
  ],
31
  "downsample_padding": 1,
32
- "dropout": 0.0,
33
  "dual_cross_attention": false,
34
  "encoder_hid_dim": null,
35
  "encoder_hid_dim_type": null,
@@ -39,7 +37,7 @@
39
  "layers_per_block": 2,
40
  "mid_block_only_cross_attention": null,
41
  "mid_block_scale_factor": 1,
42
- "mid_block_type": "UNetMidBlock2DCrossAttn",
43
  "norm_eps": 1e-05,
44
  "norm_num_groups": 32,
45
  "num_attention_heads": null,
@@ -57,9 +55,14 @@
57
  "time_embedding_type": "positional",
58
  "timestep_post_act": null,
59
  "transformer_layers_per_block": [
60
- 1,
61
- 2,
62
- 10
 
 
 
 
 
63
  ],
64
  "up_block_types": [
65
  "CrossAttnUpBlock2D",
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.19.0.dev0",
4
  "act_fn": "silu",
5
  "addition_embed_type": "text_time",
6
  "addition_embed_type_num_heads": 64,
 
10
  10,
11
  20
12
  ],
 
13
  "block_out_channels": [
14
  320,
15
  640,
 
28
  "CrossAttnDownBlock2D"
29
  ],
30
  "downsample_padding": 1,
 
31
  "dual_cross_attention": false,
32
  "encoder_hid_dim": null,
33
  "encoder_hid_dim_type": null,
 
37
  "layers_per_block": 2,
38
  "mid_block_only_cross_attention": null,
39
  "mid_block_scale_factor": 1,
40
+ "mid_block_type": "UNetMidBlock2D",
41
  "norm_eps": 1e-05,
42
  "norm_num_groups": 32,
43
  "num_attention_heads": null,
 
55
  "time_embedding_type": "positional",
56
  "timestep_post_act": null,
57
  "transformer_layers_per_block": [
58
+ [1],
59
+ [2,2],
60
+ [4,4]
61
+ ],
62
+ "reverse_transformer_layers_per_block": [
63
+ [4,4,10],
64
+ [2,1,1],
65
+ 1
66
  ],
67
  "up_block_types": [
68
  "CrossAttnUpBlock2D",