File size: 1,164 Bytes
9e1bd8d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | {
"_class_name": "AutoencoderKLLegacy",
"_diffusers_version": "0.32.2",
"causal_decoder": false,
"causal_encoder": true,
"ch": 128,
"ch_mult": [
1,
2,
2,
4,
4,
8
],
"embed_dim": 24,
"in_channels": 3,
"num_res_blocks": 2,
"num_res_blocks_decoder": null,
"out_ch": 3,
"padding_mode": "reflect",
"padding_mode_t": null,
"pixel_norm_type": "imagenet",
"scaling_factor": 1.0,
"shift_factor": 0.0,
"space_down": [
2,
2,
2,
2,
1,
1
],
"space_up": [
1,
2,
2,
2,
2,
1
],
"time_down": [
1,
2,
2,
1,
1,
1
],
"time_up": null,
"use_3d_conv": true,
"use_t_isolated_gn": true,
"use_vit_decoder": true,
"vae_ratio": 16,
"vae_ratio_t": 4,
"vit_decoder_kwargs": {
"dim_head": 64,
"ffn_activation_fn": "silu",
"ffn_use_gated": true,
"heads": 32,
"norm_affine": true,
"norm_type": "rms_norm",
"num_layers": 36,
"qk_norm_affine": false,
"qk_norm_type": "rms_norm",
"rope_dim_ratio": 0.75,
"rope_theta": 100.0
},
"z_channels": 24,
"zq_ch_decoder": null,
"zq_ch_encoder": null
}
|