Update config.json
Browse files- config.json +4 -19
config.json
CHANGED
|
@@ -6,20 +6,10 @@
|
|
| 6 |
"add_attention": true,
|
| 7 |
"attention_head_dim": null,
|
| 8 |
"attn_norm_num_groups": null,
|
| 9 |
-
"block_out_channels": [
|
| 10 |
-
128,
|
| 11 |
-
256,
|
| 12 |
-
256,
|
| 13 |
-
256
|
| 14 |
-
],
|
| 15 |
"center_input_sample": false,
|
| 16 |
"class_embed_type": null,
|
| 17 |
-
"down_block_types": [
|
| 18 |
-
"DownBlock2D",
|
| 19 |
-
"AttnDownBlock2D",
|
| 20 |
-
"DownBlock2D",
|
| 21 |
-
"DownBlock2D"
|
| 22 |
-
],
|
| 23 |
"downsample_padding": 0,
|
| 24 |
"downsample_type": "conv",
|
| 25 |
"dropout": 0.0,
|
|
@@ -36,11 +26,6 @@
|
|
| 36 |
"resnet_time_scale_shift": "default",
|
| 37 |
"sample_size": 32,
|
| 38 |
"time_embedding_type": "positional",
|
| 39 |
-
"up_block_types": [
|
| 40 |
-
"UpBlock2D",
|
| 41 |
-
"UpBlock2D",
|
| 42 |
-
"AttnUpBlock2D",
|
| 43 |
-
"UpBlock2D"
|
| 44 |
-
],
|
| 45 |
"upsample_type": "conv"
|
| 46 |
-
}
|
|
|
|
| 6 |
"add_attention": true,
|
| 7 |
"attention_head_dim": null,
|
| 8 |
"attn_norm_num_groups": null,
|
| 9 |
+
"block_out_channels": [128, 256, 256, 256],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
"center_input_sample": false,
|
| 11 |
"class_embed_type": null,
|
| 12 |
+
"down_block_types": ["DownBlock2D", "AttnDownBlock2D", "DownBlock2D", "DownBlock2D"],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
"downsample_padding": 0,
|
| 14 |
"downsample_type": "conv",
|
| 15 |
"dropout": 0.0,
|
|
|
|
| 26 |
"resnet_time_scale_shift": "default",
|
| 27 |
"sample_size": 32,
|
| 28 |
"time_embedding_type": "positional",
|
| 29 |
+
"up_block_types": ["UpBlock2D", "UpBlock2D", "AttnUpBlock2D", "UpBlock2D"],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
"upsample_type": "conv"
|
| 31 |
+
}
|