patrickvonplaten commited on
Commit
31ff4ed
1 Parent(s): a1f6625

add new config

Browse files
Files changed (1) hide show
  1. vqvae/config.json +9 -22
vqvae/config.json CHANGED
@@ -1,25 +1,12 @@
1
  {
2
- "_class_name": "AutoencoderKL",
 
3
  "_diffusers_version": "0.0.4",
4
- "attn_resolutions": [],
5
- "ch": 128,
6
- "ch_mult": [
7
- 1,
8
- 2,
9
- 4,
10
- 4
11
- ],
12
- "double_z": true,
13
- "dropout": 0.0,
14
- "embed_dim": 4,
15
- "give_pre_end": false,
16
- "in_channels": 3,
17
- "name_or_path": "fusing/latent-diffusion-text2im-large",
18
- "num_res_blocks": 2,
19
- "out_ch": 3,
20
- "remap": null,
21
- "resamp_with_conv": true,
22
- "resolution": 256,
23
- "sane_index_shape": false,
24
- "z_channels": 4
25
  }
 
1
  {
2
+ "_class_name": "VQModel",
3
+ "sample_size": 256,
4
  "_diffusers_version": "0.0.4",
5
+ "in_channels": 3,
6
+ "out_channels": 3,
7
+ "latent_channels": 4,
8
+ "down_block_types": ["DownEncoderBlock2D", "DownEncoderBlock2D", "DownEncoderBlock2D", "DownEncoderBlock2D"],
9
+ "up_block_types": ["UpDecoderBlock2D", "UpDecoderBlock2D", "UpDecoderBlock2D", "UpDecoderBlock2D"],
10
+ "block_out_channels": [128, 256, 512, 512],
11
+ "layers_per_block": 2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  }