Update vae/config.json
Browse files- vae/config.json +2 -2
vae/config.json
CHANGED
@@ -6,8 +6,8 @@
|
|
6 |
"block_out_channels": [
|
7 |
128,
|
8 |
256,
|
9 |
-
|
10 |
-
|
11 |
],
|
12 |
"down_block_types": [
|
13 |
"DownEncoderBlock2D",
|
|
|
6 |
"block_out_channels": [
|
7 |
128,
|
8 |
256,
|
9 |
+
512,
|
10 |
+
512
|
11 |
],
|
12 |
"down_block_types": [
|
13 |
"DownEncoderBlock2D",
|