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