estellea commited on
Commit
656fb8f
1 Parent(s): e7b25a2

Upload config.json

Browse files
Files changed (1) hide show
  1. vae/config.json +3 -3
vae/config.json CHANGED
@@ -14,11 +14,11 @@
14
  "DownEncoderBlock2D",
15
  "DownEncoderBlock2D"
16
  ],
17
- "in_channels": 3,
18
  "latent_channels": 4,
19
  "layers_per_block": 2,
20
- "out_channels": 3,
21
- "sample_size": 512,
22
  "scaling_factor": 0.18215,
23
  "up_block_types": [
24
  "UpDecoderBlock2D",
 
14
  "DownEncoderBlock2D",
15
  "DownEncoderBlock2D"
16
  ],
17
+ "in_channels": 6,
18
  "latent_channels": 4,
19
  "layers_per_block": 2,
20
+ "out_channels": 6,
21
+ "sample_size": 256,
22
  "scaling_factor": 0.18215,
23
  "up_block_types": [
24
  "UpDecoderBlock2D",