From 21ed4a2242b2eddd9f592333ac207c15543a987c Mon Sep 17 00:00:00 2001 From: azazeal04 <132445160+azazeal04@users.noreply.github.com> Date: Sat, 4 Apr 2026 15:55:09 +0200 Subject: [PATCH] Add TwinFlow_Z_Image class to model_base.py Added TwinFlow_Z_Image class for new model integration. --- comfy/model_base.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/comfy/model_base.py b/comfy/model_base.py index c2ae646aa..7b668d147 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -40,6 +40,7 @@ import comfy.ldm.hunyuan_video.model import comfy.ldm.cosmos.model import comfy.ldm.cosmos.predict2 import comfy.ldm.lumina.model +import comfy.ldm.twinflow.model import comfy.ldm.wan.model import comfy.ldm.wan.model_animate import comfy.ldm.hunyuan3d.model @@ -1281,6 +1282,11 @@ class ZImagePixelSpace(Lumina2): BaseModel.__init__(self, model_config, model_type, device=device, unet_model=comfy.ldm.lumina.model.NextDiTPixelSpace) self.memory_usage_factor_conds = ("ref_latents",) +class TwinFlow_Z_Image(Lumina2): + def __init__(self, model_config, model_type=ModelType.FLOW, device=None): + BaseModel.__init__(self, model_config, model_type, device=device, unet_model=comfy.ldm.twinflow.model.TwinFlowZImageTransformer) + self.memory_usage_factor_conds = ("ref_latents",) + class WAN21(BaseModel): def __init__(self, model_config, model_type=ModelType.FLOW, image_to_video=False, device=None): super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.wan.model.WanModel)