Manseo commited on
Commit
ec2885e
·
1 Parent(s): ac61493

Update vae/config.json

Browse files
Files changed (1) hide show
  1. vae/config.json +5 -6
vae/config.json CHANGED
@@ -1,12 +1,12 @@
1
- {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.12.0.dev0",
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
7
  256,
8
- 768,
9
- 768
10
  ],
11
  "down_block_types": [
12
  "DownEncoderBlock2D",
@@ -19,8 +19,7 @@
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
- "sample_size": 768,
23
- "scaling_factor": 0.18215,
24
  "up_block_types": [
25
  "UpDecoderBlock2D",
26
  "UpDecoderBlock2D",
 
1
+
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.10.2",
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
7
  256,
8
+ 512,
9
+ 512
10
  ],
11
  "down_block_types": [
12
  "DownEncoderBlock2D",
 
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
+ "sample_size": 256,
 
23
  "up_block_types": [
24
  "UpDecoderBlock2D",
25
  "UpDecoderBlock2D",