Allegro / vae /config.json
a-r-r-o-w's picture
a-r-r-o-w HF staff
add model_index.json; update configs
43d0668
raw
history blame
1.11 kB
{
"_class_name": "AutoencoderKLAllegro",
"_diffusers_version": "0.31.0.dev0",
"act_fn": "silu",
"block_out_channels": [
128,
256,
512,
512
],
"blocks_tempdown_li": [
true,
true,
false,
false
],
"blocks_tempup_li": [
false,
true,
true,
false
],
"chunk_len": 24,
"down_block_num": 4,
"down_block_types": [
"AllegroDownBlock3D",
"AllegroDownBlock3D",
"AllegroDownBlock3D",
"AllegroDownBlock3D"
],
"force_upcast": true,
"in_channels": 3,
"latent_channels": 4,
"layers_per_block": 2,
"load_mode": "full",
"norm_num_groups": 32,
"out_channels": 3,
"sample_size": 320,
"scale_factor": 0.13,
"scaling_factor": 0.13235,
"t_over": 8,
"temporal_compression_ratio": 4,
"temporal_downsample_blocks": [
true,
true,
false,
false
],
"temporal_upsample_blocks": [
false,
true,
true,
false
],
"tile_overlap": [
120,
80
],
"up_block_num": 4,
"up_block_types": [
"AllegroUpBlock3D",
"AllegroUpBlock3D",
"AllegroUpBlock3D",
"AllegroUpBlock3D"
]
}