brezende commited on
Commit
f0e468e
1 Parent(s): 37c0266

Update vae/config.json

Browse files
Files changed (1) hide show
  1. vae/config.json +2 -2
vae/config.json CHANGED
@@ -6,8 +6,8 @@
6
  "block_out_channels": [
7
  128,
8
  256,
9
- 1024,
10
- 1024
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",