patrickvonplaten commited on
Commit
63b33cf
1 Parent(s): 93bbe59

add new config

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