Update vae/config.json
Browse files- vae/config.json +1 -32
vae/config.json
CHANGED
@@ -1,32 +1 @@
|
|
1 |
-
{
|
2 |
-
"_class_name": "AutoencoderKLLTX",
|
3 |
-
"_diffusers_version": "0.32.0.dev0",
|
4 |
-
"block_out_channels": [
|
5 |
-
128,
|
6 |
-
256,
|
7 |
-
512,
|
8 |
-
512
|
9 |
-
],
|
10 |
-
"decoder_causal": false,
|
11 |
-
"encoder_causal": true,
|
12 |
-
"in_channels": 3,
|
13 |
-
"latent_channels": 128,
|
14 |
-
"layers_per_block": [
|
15 |
-
4,
|
16 |
-
3,
|
17 |
-
3,
|
18 |
-
3,
|
19 |
-
4
|
20 |
-
],
|
21 |
-
"out_channels": 3,
|
22 |
-
"patch_size": 4,
|
23 |
-
"patch_size_t": 1,
|
24 |
-
"resnet_norm_eps": 1e-06,
|
25 |
-
"scaling_factor": 1.0,
|
26 |
-
"spatio_temporal_scaling": [
|
27 |
-
true,
|
28 |
-
true,
|
29 |
-
true,
|
30 |
-
false
|
31 |
-
]
|
32 |
-
}
|
|
|
1 |
+
{"_class_name": "CausalVideoAutoencoder", "dims": 3, "in_channels": 3, "out_channels": 3, "latent_channels": 128, "blocks": [["res_x", 4], ["compress_all", 1], ["res_x_y", 1], ["res_x", 3], ["compress_all", 1], ["res_x_y", 1], ["res_x", 3], ["compress_all", 1], ["res_x", 3], ["res_x", 4]], "scaling_factor": 1.0, "norm_layer": "pixel_norm", "patch_size": 4, "latent_log_var": "uniform", "use_quant_conv": false, "causal_decoder": false}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|