SDV VideoDecoder

2023-11-22 20:20:32 +01:00
parent a3e1804def
commit 06fdd1f2d6
6 changed files with 974 additions and 0 deletions
@@ -110,5 +110,21 @@ vae_conf = {
 		"type" : "ConsistencyDecoder",
 		"embed_scale"      : 8,
 		"embed_dim"        : 4,
+	},
+	# SAI Video Decoder
+	"SDV-VideoDecoder": {
+		"type"             : "AutoencoderKL-VideoDecoder",
+		"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" : [],
+		"video_kernel_size": [3, 1, 1]
 	}
 }