Add ostris/vae-kl-f8-d16

https://huggingface.co/ostris/vae-kl-f8-d16
2024-07-02 23:02:31 +02:00
parent e866b5b6cb
commit 62b84531f4
1 changed files with 14 additions and 0 deletions
@@ -31,6 +31,20 @@ vae_conf = {
 		"num_res_blocks"   : 2,
 		"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": {
 		"type"             : "AutoencoderKL",
 		"embed_scale"      : 16,