Commit
·
420c385
1
Parent(s):
a4d5f5f
Update config.json
Browse files- 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,
|