Upload config.json
Browse files- unet/config.json +17 -0
unet/config.json
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "Kandinsky3UNet",
|
3 |
+
"_diffusers_version": "0.26.2",
|
4 |
+
"attention_head_dim": 64,
|
5 |
+
"block_out_channels": [
|
6 |
+
384,
|
7 |
+
768,
|
8 |
+
1536,
|
9 |
+
3072
|
10 |
+
],
|
11 |
+
"cross_attention_dim": 4096,
|
12 |
+
"encoder_hid_dim": 4096,
|
13 |
+
"groups": 32,
|
14 |
+
"in_channels": 4,
|
15 |
+
"layers_per_block": 3,
|
16 |
+
"time_embedding_dim": 1536
|
17 |
+
}
|