Commit ·
33fd470
1
Parent(s): a0ffc06
Update config.json
Browse files- config.json +2 -3
config.json
CHANGED
|
@@ -45,7 +45,6 @@
|
|
| 45 |
"skip_rescale": true,
|
| 46 |
"block_channels": [128, 128, 256, 256, 256, 256, 256],
|
| 47 |
"in_channels": 3,
|
| 48 |
-
"num_res_blocks": 1,
|
| 49 |
"out_channels": 3,
|
| 50 |
"time_embedding_type": "fourier",
|
| 51 |
"resnet_eps": 1e-6,
|
|
@@ -56,14 +55,14 @@
|
|
| 56 |
"UNetResSkipDownBlock2D",
|
| 57 |
"UNetResSkipDownBlock2D",
|
| 58 |
"UNetResSkipDownBlock2D",
|
| 59 |
-
"UNetResSkipDownBlock2D",
|
| 60 |
"UNetResAttnSkipDownBlock2D",
|
|
|
|
| 61 |
"UNetResSkipDownBlock2D"
|
| 62 |
],
|
| 63 |
"up_blocks": [
|
| 64 |
"UNetResSkipUpBlock2D",
|
| 65 |
-
"UNetResAttnSkipUpBlock2D",
|
| 66 |
"UNetResSkipUpBlock2D",
|
|
|
|
| 67 |
"UNetResSkipUpBlock2D",
|
| 68 |
"UNetResSkipUpBlock2D",
|
| 69 |
"UNetResSkipUpBlock2D",
|
|
|
|
| 45 |
"skip_rescale": true,
|
| 46 |
"block_channels": [128, 128, 256, 256, 256, 256, 256],
|
| 47 |
"in_channels": 3,
|
|
|
|
| 48 |
"out_channels": 3,
|
| 49 |
"time_embedding_type": "fourier",
|
| 50 |
"resnet_eps": 1e-6,
|
|
|
|
| 55 |
"UNetResSkipDownBlock2D",
|
| 56 |
"UNetResSkipDownBlock2D",
|
| 57 |
"UNetResSkipDownBlock2D",
|
|
|
|
| 58 |
"UNetResAttnSkipDownBlock2D",
|
| 59 |
+
"UNetResSkipDownBlock2D",
|
| 60 |
"UNetResSkipDownBlock2D"
|
| 61 |
],
|
| 62 |
"up_blocks": [
|
| 63 |
"UNetResSkipUpBlock2D",
|
|
|
|
| 64 |
"UNetResSkipUpBlock2D",
|
| 65 |
+
"UNetResAttnSkipUpBlock2D",
|
| 66 |
"UNetResSkipUpBlock2D",
|
| 67 |
"UNetResSkipUpBlock2D",
|
| 68 |
"UNetResSkipUpBlock2D",
|