Upload config.json
Browse filesUpdate the unet config for the cd_imagenet64 consistency models checkpoint for slow testing (fix mid block group norm).
diffusers_cd_imagenet64_l2/config.json
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
{
|
2 |
"_class_name": "UNet2DModel",
|
3 |
-
"_diffusers_version": "0.
|
4 |
"act_fn": "silu",
|
5 |
"add_attention": true,
|
6 |
"attention_head_dim": 64,
|
|
|
7 |
"block_out_channels": [
|
8 |
192,
|
9 |
384,
|
@@ -19,6 +20,7 @@
|
|
19 |
"AttnDownBlock2D"
|
20 |
],
|
21 |
"downsample_padding": 1,
|
|
|
22 |
"flip_sin_to_cos": true,
|
23 |
"freq_shift": 0,
|
24 |
"in_channels": 3,
|
@@ -31,12 +33,11 @@
|
|
31 |
"resnet_time_scale_shift": "scale_shift",
|
32 |
"sample_size": 64,
|
33 |
"time_embedding_type": "positional",
|
34 |
-
"upsample_type": "resnet",
|
35 |
-
"downsample_type": "resnet",
|
36 |
"up_block_types": [
|
37 |
"AttnUpBlock2D",
|
38 |
"AttnUpBlock2D",
|
39 |
"AttnUpBlock2D",
|
40 |
"ResnetUpsampleBlock2D"
|
41 |
-
]
|
|
|
42 |
}
|
|
|
1 |
{
|
2 |
"_class_name": "UNet2DModel",
|
3 |
+
"_diffusers_version": "0.21.0.dev0",
|
4 |
"act_fn": "silu",
|
5 |
"add_attention": true,
|
6 |
"attention_head_dim": 64,
|
7 |
+
"attn_norm_num_groups": 32,
|
8 |
"block_out_channels": [
|
9 |
192,
|
10 |
384,
|
|
|
20 |
"AttnDownBlock2D"
|
21 |
],
|
22 |
"downsample_padding": 1,
|
23 |
+
"downsample_type": "resnet",
|
24 |
"flip_sin_to_cos": true,
|
25 |
"freq_shift": 0,
|
26 |
"in_channels": 3,
|
|
|
33 |
"resnet_time_scale_shift": "scale_shift",
|
34 |
"sample_size": 64,
|
35 |
"time_embedding_type": "positional",
|
|
|
|
|
36 |
"up_block_types": [
|
37 |
"AttnUpBlock2D",
|
38 |
"AttnUpBlock2D",
|
39 |
"AttnUpBlock2D",
|
40 |
"ResnetUpsampleBlock2D"
|
41 |
+
],
|
42 |
+
"upsample_type": "resnet"
|
43 |
}
|