sucess add dcae into the repo;

2024-11-30 05:11:01 -08:00
parent c6527a45f1
commit 861a378edf
3 changed files with 1041 additions and 0 deletions
@@ -155,5 +155,25 @@ vae_conf = {
 		"ch_mult"          : [1, 2, 2, 4],
 		"num_res_blocks"   : 2,
 		"attn_resolutions" : [32],
+    },
+    # DCAE configs
+    "dcae-f32c32-sana-1.0": {
+        "type"					: "DCAE",
+        "in_channels"			: 3,
+		"embed_scale"      		: 32,
+        "embed_dim"				: 32,
+        "encoder_block_type"	: ["ResBlock", "ResBlock", "ResBlock", "EViTS5GLU", "EViTS5GLU", "EViTS5GLU"],
+        "encoder_width_list"	: [128, 256, 512, 512, 1024, 1024],
+        "encoder_depth_list"	: [2, 2, 2, 3, 3, 3],
+        "encoder_norm"			: "rms2d",
+        "encoder_act"			: "silu",
+        "downsample_block_type"	: "Conv",
+        "decoder_block_type"	: ["ResBlock", "ResBlock", "ResBlock", "EViTS5GLU", "EViTS5GLU", "EViTS5GLU"],
+        "decoder_width_list"	: [128, 256, 512, 512, 1024, 1024],
+        "decoder_depth_list"	: [3, 3, 3, 3, 3, 3],
+        "decoder_norm"			: "rms2d", 
+        "decoder_act"			: "silu",
+        "upsample_block_type"	: "InterpolateConv",
+        "scaling_factor"		: 0.41407
    }
 }
@@ -32,6 +32,9 @@ class EXVAE(comfy.sd.VAE):
 		elif model_conf["type"] == "MoVQ3":
 			from .models.movq3 import MoVQ
 			model = MoVQ(model_conf)
+		elif model_conf["type"] == "DCAE":
+			from .models.dcae import DCAE
+			model = DCAE(**model_conf)
 		else:
 			raise NotImplementedError(f"Unknown VAE type '{model_conf['type']}'")