Gjm1234 commited on
Commit
b03e669
·
verified ·
1 Parent(s): ea3c6d3

Update vae/config.json

Browse files
Files changed (1) hide show
  1. vae/config.json +17 -10
vae/config.json CHANGED
@@ -1,17 +1,24 @@
1
  {
2
  "_class_name": "AutoencoderKL",
 
3
  "act_fn": "silu",
4
- "block_out_channels": [
5
- 128,
6
- 256,
7
- 512,
8
- 512
9
- ],
10
  "latent_channels": 4,
11
- "norm_num_groups": 32,
12
  "sample_size": 512,
13
  "scaling_factor": 0.18215,
14
- "in_channels": 3,
15
- "out_channels": 3,
16
- "norm_type": "layer_norm"
 
 
 
 
 
 
 
 
 
 
17
  }
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.30.0",
4
  "act_fn": "silu",
5
+ "block_out_channels": [128, 256, 512, 512],
6
+ "in_channels": 3,
7
+ "out_channels": 3,
 
 
 
8
  "latent_channels": 4,
 
9
  "sample_size": 512,
10
  "scaling_factor": 0.18215,
11
+ "norm_num_groups": 32,
12
+ "down_block_types": [
13
+ "DownEncoderBlock2D",
14
+ "DownEncoderBlock2D",
15
+ "DownEncoderBlock2D",
16
+ "DownEncoderBlock2D"
17
+ ],
18
+ "up_block_types": [
19
+ "UpDecoderBlock2D",
20
+ "UpDecoderBlock2D",
21
+ "UpDecoderBlock2D",
22
+ "UpDecoderBlock2D"
23
+ ]
24
  }