wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-18 22:42:35 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Add functions to check and extract TwinFlow-Z-Image config

Browse Source
This commit is contained in:
azazeal04 2026-04-04 15:59:50 +02:00 committed by GitHub
parent 84c983a779
commit dc413071ff
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comfy/sd.py
View File

@ -74,6 +74,20 @@ import comfy.latent_formats
import comfy.ldm.flux.redux import comfy.ldm.flux.redux
def is_twinflow_z_image_model(state_dict):
"""Check if model state dict is TwinFlow-Z-Image."""
return any(k.startswith("t_embedder_2.") for k in state_dict)
def get_twinflow_z_image_config(state_dict):
"""Extract TwinFlow-Z-Image configuration from state dict."""
if not is_twinflow_z_image_model(state_dict):
return {}
return {
"image_model": "twinflow_z_image",
"architecture": "TwinFlow_Z_Image",
}
def load_lora_for_models(model, clip, lora, strength_model, strength_clip): def load_lora_for_models(model, clip, lora, strength_model, strength_clip):
key_map = {} key_map = {}
if model is not None: if model is not None: