Commit
·
fd221dd
1
Parent(s):
dce8392
Upload config.json (#1)
Browse files- Upload config.json (4e2aa3eeeaa8322a380eafae4a89ed1f1734df96)
Co-authored-by: Daniel Gu <dg845@users.noreply.huggingface.co>
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 |
}
|