Update unet/config.json
Browse files- unet/config.json +6 -3
unet/config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"_class_name": "UNet2DConditionModel",
|
| 3 |
-
"_diffusers_version": "0.
|
| 4 |
"act_fn": "silu",
|
| 5 |
"attention_head_dim": 8,
|
| 6 |
"block_out_channels": [
|
|
@@ -10,6 +10,7 @@
|
|
| 10 |
1280
|
| 11 |
],
|
| 12 |
"center_input_sample": false,
|
|
|
|
| 13 |
"cross_attention_dim": 768,
|
| 14 |
"down_block_types": [
|
| 15 |
"CrossAttnDownBlock2D",
|
|
@@ -24,12 +25,14 @@
|
|
| 24 |
"in_channels": 4,
|
| 25 |
"layers_per_block": 2,
|
| 26 |
"mid_block_scale_factor": 1,
|
|
|
|
| 27 |
"norm_eps": 1e-05,
|
| 28 |
"norm_num_groups": 32,
|
| 29 |
"num_class_embeds": null,
|
| 30 |
"only_cross_attention": false,
|
| 31 |
"out_channels": 4,
|
| 32 |
-
"
|
|
|
|
| 33 |
"up_block_types": [
|
| 34 |
"UpBlock2D",
|
| 35 |
"CrossAttnUpBlock2D",
|
|
@@ -38,4 +41,4 @@
|
|
| 38 |
],
|
| 39 |
"upcast_attention": false,
|
| 40 |
"use_linear_projection": false
|
| 41 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
"_class_name": "UNet2DConditionModel",
|
| 3 |
+
"_diffusers_version": "0.12.0.dev0",
|
| 4 |
"act_fn": "silu",
|
| 5 |
"attention_head_dim": 8,
|
| 6 |
"block_out_channels": [
|
|
|
|
| 10 |
1280
|
| 11 |
],
|
| 12 |
"center_input_sample": false,
|
| 13 |
+
"class_embed_type": null,
|
| 14 |
"cross_attention_dim": 768,
|
| 15 |
"down_block_types": [
|
| 16 |
"CrossAttnDownBlock2D",
|
|
|
|
| 25 |
"in_channels": 4,
|
| 26 |
"layers_per_block": 2,
|
| 27 |
"mid_block_scale_factor": 1,
|
| 28 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
| 29 |
"norm_eps": 1e-05,
|
| 30 |
"norm_num_groups": 32,
|
| 31 |
"num_class_embeds": null,
|
| 32 |
"only_cross_attention": false,
|
| 33 |
"out_channels": 4,
|
| 34 |
+
"resnet_time_scale_shift": "default",
|
| 35 |
+
"sample_size": 64,
|
| 36 |
"up_block_types": [
|
| 37 |
"UpBlock2D",
|
| 38 |
"CrossAttnUpBlock2D",
|
|
|
|
| 41 |
],
|
| 42 |
"upcast_attention": false,
|
| 43 |
"use_linear_projection": false
|
| 44 |
+
}
|