+14
@@ -31,6 +31,20 @@ vae_conf = {
|
|||||||
"num_res_blocks" : 2,
|
"num_res_blocks" : 2,
|
||||||
"attn_resolutions" : [],
|
"attn_resolutions" : [],
|
||||||
},
|
},
|
||||||
|
"kl-f8-d16": { # 16 channel VAE from https://huggingface.co/ostris/vae-kl-f8-d16/tree/main
|
||||||
|
"type" : "AutoencoderKL",
|
||||||
|
"embed_scale" : 8,
|
||||||
|
"embed_dim" : 16,
|
||||||
|
"z_channels" : 16,
|
||||||
|
"double_z" : True,
|
||||||
|
"resolution" : 256,
|
||||||
|
"in_channels" : 3,
|
||||||
|
"out_ch" : 3,
|
||||||
|
"ch" : 128,
|
||||||
|
"ch_mult" : [1,1,2,4],
|
||||||
|
"num_res_blocks" : 2,
|
||||||
|
"attn_resolutions" : [],
|
||||||
|
},
|
||||||
"kl-f16": {
|
"kl-f16": {
|
||||||
"type" : "AutoencoderKL",
|
"type" : "AutoencoderKL",
|
||||||
"embed_scale" : 16,
|
"embed_scale" : 16,
|
||||||
|
|||||||
Reference in New Issue
Block a user