patrickvonplaten commited on
Commit
518665a
·
1 Parent(s): f9efe66

add new config

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