Upload config.json
Browse files- config.json +32 -27
config.json
CHANGED
@@ -1,32 +1,37 @@
|
|
1 |
{
|
2 |
-
"_class_name": "
|
3 |
-
"_diffusers_version": "0.
|
4 |
-
"
|
5 |
-
|
6 |
-
|
|
|
|
|
|
|
|
|
7 |
],
|
8 |
-
"
|
9 |
-
|
10 |
-
|
|
|
|
|
|
|
|
|
11 |
],
|
12 |
-
"
|
13 |
-
|
14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
],
|
16 |
-
"
|
17 |
-
|
18 |
-
"CLIPTextModel"
|
19 |
-
],
|
20 |
-
"tokenizer": [
|
21 |
-
"transformers",
|
22 |
-
"CLIPTokenizer"
|
23 |
-
],
|
24 |
-
"unet": [
|
25 |
-
"diffusers",
|
26 |
-
"UNet2DConditionModel"
|
27 |
-
],
|
28 |
-
"vae": [
|
29 |
-
"diffusers",
|
30 |
-
"AutoencoderKL"
|
31 |
-
]
|
32 |
}
|
|
|
1 |
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.10.2",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"attention_head_dim": 64,
|
6 |
+
"block_out_channels": [
|
7 |
+
128,
|
8 |
+
256,
|
9 |
+
512,
|
10 |
+
1024
|
11 |
],
|
12 |
+
"center_input_sample": false,
|
13 |
+
"cross_attention_dim": 2048,
|
14 |
+
"down_block_types": [
|
15 |
+
"CrossAttnDownBlock2D",
|
16 |
+
"CrossAttnDownBlock2D",
|
17 |
+
"CrossAttnDownBlock2D",
|
18 |
+
"DownBlock2D"
|
19 |
],
|
20 |
+
"dropout": 0.0,
|
21 |
+
"flip_sin_to_cos": true,
|
22 |
+
"freq_shift": 0,
|
23 |
+
"in_channels": 4,
|
24 |
+
"layers_per_block": 2,
|
25 |
+
"norm_eps": 1e-05,
|
26 |
+
"norm_num_groups": 32,
|
27 |
+
"out_channels": 4,
|
28 |
+
"sample_size": 1024,
|
29 |
+
"up_block_types": [
|
30 |
+
"UpBlock2D",
|
31 |
+
"CrossAttnUpBlock2D",
|
32 |
+
"CrossAttnUpBlock2D",
|
33 |
+
"CrossAttnUpBlock2D"
|
34 |
],
|
35 |
+
"use_linear_projection": true,
|
36 |
+
"use_memory_efficient_attention": true
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
}
|