Commit ·
e0aaf19
1
Parent(s): 434329b
Update unet/config.json
Browse files- unet/config.json +8 -6
unet/config.json
CHANGED
|
@@ -14,9 +14,9 @@
|
|
| 14 |
"class_embed_type": null,
|
| 15 |
"down_block_types": [
|
| 16 |
"ResnetDownsampleBlock2D",
|
| 17 |
-
"
|
| 18 |
-
"
|
| 19 |
-
"
|
| 20 |
],
|
| 21 |
"downsample_padding": 1,
|
| 22 |
"flip_sin_to_cos": true,
|
|
@@ -31,10 +31,12 @@
|
|
| 31 |
"resnet_time_scale_shift": "scale_shift",
|
| 32 |
"sample_size": 64,
|
| 33 |
"time_embedding_type": "positional",
|
|
|
|
|
|
|
| 34 |
"up_block_types": [
|
| 35 |
-
"
|
| 36 |
-
"
|
| 37 |
-
"
|
| 38 |
"ResnetUpsampleBlock2D"
|
| 39 |
]
|
| 40 |
}
|
|
|
|
| 14 |
"class_embed_type": null,
|
| 15 |
"down_block_types": [
|
| 16 |
"ResnetDownsampleBlock2D",
|
| 17 |
+
"AttnDownBlock2D",
|
| 18 |
+
"AttnDownBlock2D",
|
| 19 |
+
"AttnDownBlock2D"
|
| 20 |
],
|
| 21 |
"downsample_padding": 1,
|
| 22 |
"flip_sin_to_cos": true,
|
|
|
|
| 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 |
}
|