Update vae/config.json
Browse files- vae/config.json +53 -27
vae/config.json
CHANGED
|
@@ -1,17 +1,25 @@
|
|
| 1 |
{
|
| 2 |
-
"_class_name": "
|
| 3 |
"_diffusers_version": "0.32.0.dev0",
|
| 4 |
-
"
|
| 5 |
-
"
|
| 6 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
"ResBlock",
|
| 8 |
"ResBlock",
|
| 9 |
"ResBlock",
|
| 10 |
-
"
|
| 11 |
-
"
|
| 12 |
-
"
|
| 13 |
],
|
| 14 |
-
"
|
| 15 |
3,
|
| 16 |
3,
|
| 17 |
3,
|
|
@@ -19,8 +27,23 @@
|
|
| 19 |
3,
|
| 20 |
3
|
| 21 |
],
|
| 22 |
-
"
|
| 23 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
128,
|
| 25 |
256,
|
| 26 |
512,
|
|
@@ -28,17 +51,15 @@
|
|
| 28 |
1024,
|
| 29 |
1024
|
| 30 |
],
|
| 31 |
-
"
|
| 32 |
-
"encoder_act": "silu",
|
| 33 |
-
"encoder_block_type": [
|
| 34 |
"ResBlock",
|
| 35 |
"ResBlock",
|
| 36 |
"ResBlock",
|
| 37 |
-
"
|
| 38 |
-
"
|
| 39 |
-
"
|
| 40 |
],
|
| 41 |
-
"
|
| 42 |
2,
|
| 43 |
2,
|
| 44 |
2,
|
|
@@ -46,17 +67,22 @@
|
|
| 46 |
3,
|
| 47 |
3
|
| 48 |
],
|
| 49 |
-
"
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
],
|
| 58 |
"in_channels": 3,
|
| 59 |
"latent_channels": 32,
|
| 60 |
"scaling_factor": 0.41407,
|
| 61 |
-
"upsample_block_type": "
|
| 62 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"_class_name": "AutoencoderDC",
|
| 3 |
"_diffusers_version": "0.32.0.dev0",
|
| 4 |
+
"attention_head_dim": 32,
|
| 5 |
+
"decoder_act_fns": "silu",
|
| 6 |
+
"decoder_block_out_channels": [
|
| 7 |
+
128,
|
| 8 |
+
256,
|
| 9 |
+
512,
|
| 10 |
+
512,
|
| 11 |
+
1024,
|
| 12 |
+
1024
|
| 13 |
+
],
|
| 14 |
+
"decoder_block_types": [
|
| 15 |
"ResBlock",
|
| 16 |
"ResBlock",
|
| 17 |
"ResBlock",
|
| 18 |
+
"EfficientViTBlock",
|
| 19 |
+
"EfficientViTBlock",
|
| 20 |
+
"EfficientViTBlock"
|
| 21 |
],
|
| 22 |
+
"decoder_layers_per_block": [
|
| 23 |
3,
|
| 24 |
3,
|
| 25 |
3,
|
|
|
|
| 27 |
3,
|
| 28 |
3
|
| 29 |
],
|
| 30 |
+
"decoder_norm_types": "rms_norm",
|
| 31 |
+
"decoder_qkv_multiscales": [
|
| 32 |
+
[],
|
| 33 |
+
[],
|
| 34 |
+
[],
|
| 35 |
+
[
|
| 36 |
+
5
|
| 37 |
+
],
|
| 38 |
+
[
|
| 39 |
+
5
|
| 40 |
+
],
|
| 41 |
+
[
|
| 42 |
+
5
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"downsample_block_type": "Conv",
|
| 46 |
+
"encoder_block_out_channels": [
|
| 47 |
128,
|
| 48 |
256,
|
| 49 |
512,
|
|
|
|
| 51 |
1024,
|
| 52 |
1024
|
| 53 |
],
|
| 54 |
+
"encoder_block_types": [
|
|
|
|
|
|
|
| 55 |
"ResBlock",
|
| 56 |
"ResBlock",
|
| 57 |
"ResBlock",
|
| 58 |
+
"EfficientViTBlock",
|
| 59 |
+
"EfficientViTBlock",
|
| 60 |
+
"EfficientViTBlock"
|
| 61 |
],
|
| 62 |
+
"encoder_layers_per_block": [
|
| 63 |
2,
|
| 64 |
2,
|
| 65 |
2,
|
|
|
|
| 67 |
3,
|
| 68 |
3
|
| 69 |
],
|
| 70 |
+
"encoder_qkv_multiscales": [
|
| 71 |
+
[],
|
| 72 |
+
[],
|
| 73 |
+
[],
|
| 74 |
+
[
|
| 75 |
+
5
|
| 76 |
+
],
|
| 77 |
+
[
|
| 78 |
+
5
|
| 79 |
+
],
|
| 80 |
+
[
|
| 81 |
+
5
|
| 82 |
+
]
|
| 83 |
],
|
| 84 |
"in_channels": 3,
|
| 85 |
"latent_channels": 32,
|
| 86 |
"scaling_factor": 0.41407,
|
| 87 |
+
"upsample_block_type": "interpolate"
|
| 88 |
+
}
|