Yntec commited on
Commit
7ed30c6
1 Parent(s): 2185672

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +3 -1
unet/config.json CHANGED
@@ -4,6 +4,7 @@
4
  "act_fn": "silu",
5
  "addition_embed_type": null,
6
  "addition_embed_type_num_heads": 64,
 
7
  "attention_head_dim": 8,
8
  "block_out_channels": [
9
  320,
@@ -51,6 +52,7 @@
51
  "time_embedding_dim": null,
52
  "time_embedding_type": "positional",
53
  "timestep_post_act": null,
 
54
  "up_block_types": [
55
  "UpBlock2D",
56
  "CrossAttnUpBlock2D",
@@ -59,4 +61,4 @@
59
  ],
60
  "upcast_attention": null,
61
  "use_linear_projection": false
62
- }
 
4
  "act_fn": "silu",
5
  "addition_embed_type": null,
6
  "addition_embed_type_num_heads": 64,
7
+ "addition_time_embed_dim": null,
8
  "attention_head_dim": 8,
9
  "block_out_channels": [
10
  320,
 
52
  "time_embedding_dim": null,
53
  "time_embedding_type": "positional",
54
  "timestep_post_act": null,
55
+ "transformer_layers_per_block": 1,
56
  "up_block_types": [
57
  "UpBlock2D",
58
  "CrossAttnUpBlock2D",
 
61
  ],
62
  "upcast_attention": null,
63
  "use_linear_projection": false
64
+ }