sucess add dcae into the repo;

This commit is contained in:
junsong
2024-11-30 05:11:01 -08:00
parent c6527a45f1
commit 861a378edf
3 changed files with 1041 additions and 0 deletions
+3
View File
@@ -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']}'")