config.json
842 B · 45 lines · json Raw
1 {
2 "_class_name": "AutoencoderKLFlux2",
3 "_diffusers_version": "0.32.0.dev0",
4 "in_channels": 3,
5 "out_channels": 3,
6 "down_block_types": [
7 "DownEncoderBlock2D",
8 "DownEncoderBlock2D",
9 "DownEncoderBlock2D",
10 "DownEncoderBlock2D"
11 ],
12 "up_block_types": [
13 "UpDecoderBlock2D",
14 "UpDecoderBlock2D",
15 "UpDecoderBlock2D",
16 "UpDecoderBlock2D"
17 ],
18 "block_out_channels": [
19 128,
20 256,
21 512,
22 512
23 ],
24 "decoder_block_out_channels": [
25 96,
26 192,
27 384,
28 384
29 ],
30 "layers_per_block": 2,
31 "act_fn": "silu",
32 "latent_channels": 32,
33 "norm_num_groups": 32,
34 "sample_size": 1024,
35 "force_upcast": true,
36 "use_quant_conv": true,
37 "use_post_quant_conv": true,
38 "mid_block_add_attention": true,
39 "batch_norm_eps": 0.0001,
40 "batch_norm_momentum": 0.1,
41 "patch_size": [
42 2,
43 2
44 ]
45 }