Custom VAE

2023-10-21 21:26:23 +02:00
parent bca8487272
commit 1c72bd4ae7
6 changed files with 680 additions and 1 deletions
@@ -0,0 +1,61 @@
+"""
+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],
+	},
+}