Sapir commited on
Commit
c7c8ad4
1 Parent(s): 07ef7f7

Update vae/config.json

Browse files
Files changed (1) hide show
  1. 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}