patrickvonplaten commited on
Commit
420c385
·
1 Parent(s): a4d5f5f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +0 -85
config.json CHANGED
@@ -2,72 +2,7 @@
2
  "_class_name": "ConsistencyDecoderVAE",
3
  "_diffusers_version": "0.23.0.dev0",
4
  "_name_or_path": "openai/consistency-decoder",
5
- "block_out_channels": [
6
- 128,
7
- 256,
8
- 512,
9
- 512
10
- ],
11
  "decoder_add_attention": false,
12
- "decoder_args": {
13
- "_use_default_values": [
14
- "downsample_type",
15
- "flip_sin_to_cos",
16
- "mid_block_scale_factor",
17
- "num_class_embeds",
18
- "attention_head_dim",
19
- "sample_size",
20
- "act_fn",
21
- "freq_shift",
22
- "class_embed_type",
23
- "attn_norm_num_groups",
24
- "downsample_padding",
25
- "upsample_type",
26
- "dropout",
27
- "center_input_sample"
28
- ],
29
- "act_fn": "silu",
30
- "add_attention": false,
31
- "attention_head_dim": 8,
32
- "attn_norm_num_groups": null,
33
- "block_out_channels": [
34
- 320,
35
- 640,
36
- 1024,
37
- 1024
38
- ],
39
- "center_input_sample": false,
40
- "class_embed_type": null,
41
- "down_block_types": [
42
- "ResnetDownsampleBlock2D",
43
- "ResnetDownsampleBlock2D",
44
- "ResnetDownsampleBlock2D",
45
- "ResnetDownsampleBlock2D"
46
- ],
47
- "downsample_padding": 1,
48
- "downsample_type": "conv",
49
- "dropout": 0.0,
50
- "flip_sin_to_cos": true,
51
- "freq_shift": 0,
52
- "in_channels": 7,
53
- "layers_per_block": 3,
54
- "mid_block_scale_factor": 1,
55
- "norm_eps": 1e-05,
56
- "norm_num_groups": 32,
57
- "num_class_embeds": null,
58
- "num_train_timesteps": 1024,
59
- "out_channels": 6,
60
- "resnet_time_scale_shift": "scale_shift",
61
- "sample_size": null,
62
- "time_embedding_type": "learned",
63
- "up_block_types": [
64
- "ResnetUpsampleBlock2D",
65
- "ResnetUpsampleBlock2D",
66
- "ResnetUpsampleBlock2D",
67
- "ResnetUpsampleBlock2D"
68
- ],
69
- "upsample_type": "conv"
70
- },
71
  "decoder_block_out_channels": [
72
  320,
73
  640,
@@ -96,26 +31,6 @@
96
  "ResnetUpsampleBlock2D"
97
  ],
98
  "encoder_act_fn": "silu",
99
- "encoder_args": {
100
- "act_fn": "silu",
101
- "block_out_channels": [
102
- 128,
103
- 256,
104
- 512,
105
- 512
106
- ],
107
- "double_z": true,
108
- "down_block_types": [
109
- "DownEncoderBlock2D",
110
- "DownEncoderBlock2D",
111
- "DownEncoderBlock2D",
112
- "DownEncoderBlock2D"
113
- ],
114
- "in_channels": 3,
115
- "layers_per_block": 2,
116
- "norm_num_groups": 32,
117
- "out_channels": 4
118
- },
119
  "encoder_block_out_channels": [
120
  128,
121
  256,
 
2
  "_class_name": "ConsistencyDecoderVAE",
3
  "_diffusers_version": "0.23.0.dev0",
4
  "_name_or_path": "openai/consistency-decoder",
 
 
 
 
 
 
5
  "decoder_add_attention": false,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  "decoder_block_out_channels": [
7
  320,
8
  640,
 
31
  "ResnetUpsampleBlock2D"
32
  ],
33
  "encoder_act_fn": "silu",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  "encoder_block_out_channels": [
35
  128,
36
  256,