62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
"""
|
|
List of all VAE configs, with training parts stripped.
|
|
"""
|
|
vae_conf = {
|
|
"kl-f4": {
|
|
"type" : "AutoencoderKL",
|
|
"embed_scale" : 4,
|
|
"embed_dim" : 3,
|
|
"z_channels" : 3,
|
|
"double_z" : True,
|
|
"resolution" : 256,
|
|
"in_channels" : 3,
|
|
"out_ch" : 3,
|
|
"ch" : 128,
|
|
"ch_mult" : [1,2,4],
|
|
"num_res_blocks" : 2,
|
|
"attn_resolutions" : [],
|
|
},
|
|
"kl-f8": { # Default SD1.5 VAE
|
|
"type" : "AutoencoderKL",
|
|
"embed_scale" : 8,
|
|
"embed_dim" : 4,
|
|
"z_channels" : 4,
|
|
"double_z" : True,
|
|
"resolution" : 256,
|
|
"in_channels" : 3,
|
|
"out_ch" : 3,
|
|
"ch" : 128,
|
|
"ch_mult" : [1,2,4,4],
|
|
"num_res_blocks" : 2,
|
|
"attn_resolutions" : [],
|
|
},
|
|
"kl-f16": {
|
|
"type" : "AutoencoderKL",
|
|
"embed_scale" : 16,
|
|
"embed_dim" : 16,
|
|
"z_channels" : 16,
|
|
"double_z" : True,
|
|
"resolution" : 256,
|
|
"in_channels" : 3,
|
|
"out_ch" : 3,
|
|
"ch" : 128,
|
|
"ch_mult" : [1,1,2,2,4],
|
|
"num_res_blocks" : 2,
|
|
"attn_resolutions" : [16],
|
|
},
|
|
"kl-f32": {
|
|
"type" : "AutoencoderKL",
|
|
"embed_scale" : 32,
|
|
"embed_dim" : 64,
|
|
"z_channels" : 64,
|
|
"double_z" : True,
|
|
"resolution" : 256,
|
|
"in_channels" : 3,
|
|
"out_ch" : 3,
|
|
"ch" : 128,
|
|
"ch_mult" : [1,1,2,2,4,4],
|
|
"num_res_blocks" : 2,
|
|
"attn_resolutions" : [16,8],
|
|
},
|
|
}
|