Added code mapping for DCAE for model: https://huggingface.co/mit-han-lab/dc-ae-f32c32-sana-1.0/blob/main/model.safetensors

2024-12-17 12:05:15 +08:00
parent b47223d5ca
commit a0cf329fda
2 changed files with 352 additions and 0 deletions
@@ -34,6 +34,8 @@ class EXVAE(comfy.sd.VAE):
 			model = MoVQ(model_conf)
 		elif model_conf["type"] == "DCAE":
 			from .models.dcae import DCAE
+			from .models import dcae_key_mapping
+			sd = dcae_key_mapping.convert_sd(sd)
 			model = DCAE(**model_conf)
 		else:
 			raise NotImplementedError(f"Unknown VAE type '{model_conf['type']}'")