Update unet/config.json
Browse files- 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 |
+
}
|