brezende commited on
Commit
0cfcbbb
1 Parent(s): 3986918

Update vae/config.json

Browse files
Files changed (1) hide show
  1. vae/config.json +2 -2
vae/config.json CHANGED
@@ -6,7 +6,7 @@
6
  "block_out_channels": [
7
  128,
8
  256,
9
- 1024,
10
  512
11
  ],
12
  "down_block_types": [
@@ -20,7 +20,7 @@
20
  "layers_per_block": 2,
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
- "sample_size": 1024,
24
  "scaling_factor": 0.18215,
25
  "up_block_types": [
26
  "UpDecoderBlock2D",
 
6
  "block_out_channels": [
7
  128,
8
  256,
9
+ 512,
10
  512
11
  ],
12
  "down_block_types": [
 
20
  "layers_per_block": 2,
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
+ "sample_size": 512,
24
  "scaling_factor": 0.18215,
25
  "up_block_types": [
26
  "UpDecoderBlock2D",