Add movq3 VAE

2024-04-06 22:13:32 +02:00
parent 18fb3f5674
commit 059c87ebb5
4 changed files with 638 additions and 1 deletions
@@ -29,6 +29,9 @@ class EXVAE(comfy.sd.VAE):
 			from .models.consistencydecoder import ConsistencyDecoder
 			model = ConsistencyDecoder()
 			sd = {f"model.{k}":v for k,v in sd.items()}
+		elif model_conf["type"] == "MoVQ3":
+			from .models.movq3 import MoVQ
+			model = MoVQ(model_conf)
 		else:
 			raise NotImplementedError(f"Unknown VAE type '{model_conf['type']}'")