Update config.json
Browse files- config.json +6 -3
config.json
CHANGED
@@ -3,12 +3,15 @@
|
|
3 |
"_diffusers_version": "0.28.2",
|
4 |
"act_fn": "silu",
|
5 |
"block_out_channels": [
|
6 |
-
32,
|
|
|
|
|
|
|
7 |
],
|
8 |
"down_block_types": [
|
9 |
"DownEncoderBlock2D",
|
10 |
"DownEncoderBlock2D",
|
11 |
-
"DownEncoderBlock2D"
|
12 |
"DownEncoderBlock2D",
|
13 |
],
|
14 |
"force_upcast": true,
|
@@ -25,6 +28,6 @@
|
|
25 |
"UpDecoderBlock2D",
|
26 |
"UpDecoderBlock2D",
|
27 |
"UpDecoderBlock2D",
|
28 |
-
"UpDecoderBlock2D"
|
29 |
]
|
30 |
}
|
|
|
3 |
"_diffusers_version": "0.28.2",
|
4 |
"act_fn": "silu",
|
5 |
"block_out_channels": [
|
6 |
+
32,
|
7 |
+
64,
|
8 |
+
128,
|
9 |
+
256
|
10 |
],
|
11 |
"down_block_types": [
|
12 |
"DownEncoderBlock2D",
|
13 |
"DownEncoderBlock2D",
|
14 |
+
"DownEncoderBlock2D"
|
15 |
"DownEncoderBlock2D",
|
16 |
],
|
17 |
"force_upcast": true,
|
|
|
28 |
"UpDecoderBlock2D",
|
29 |
"UpDecoderBlock2D",
|
30 |
"UpDecoderBlock2D",
|
31 |
+
"UpDecoderBlock2D"
|
32 |
]
|
33 |
}
|