wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 13:02:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Add twinflow_z_image_lora_to_diffusers function

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

11
comfy/lora_convert.py
View File

@ -32,7 +32,14 @@ def convert_uso_lora(sd):
sd_out[k_to] = tensor
return sd_out
def twinflow_z_image_lora_to_diffusers(state_dict):
"""Convert TwinFlow LoRA state dict for diffusers compatibility."""
for key in list(state_dict.keys()):
if "t_embedder_2" not in key and key.startswith("t_embedder."):
new_key = key.replace("t_embedder.", "t_embedder_2.", 1)
state_dict[new_key] = state_dict.pop(key)
return state_dict
def convert_lora(sd):
if "img_in.lora_A.weight" in sd and "single_blocks.0.norm.key_norm.scale" in sd:
return convert_lora_bfl_control(sd)
@ -40,4 +47,6 @@ def convert_lora(sd):
return convert_lora_wan_fun(sd)
if "single_blocks.37.processor.qkv_lora.up.weight" in sd and "double_blocks.18.processor.qkv_lora2.up.weight" in sd:
return convert_uso_lora(sd)
if any(k.startswith("t_embedder.") for k in sd.keys()):
return twinflow_z_image_lora_to_diffusers(sd)
return sd