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

Implement TwinFlow-Z-Image LoRA compatibility

Browse Source 
Add compatibility for TwinFlow-Z-Image LoRAs to target t_embedder_2.* keys.
This commit is contained in:
azazeal04 2026-04-04 15:49:32 +02:00 committed by GitHub
parent f21f6b2212
commit 00c4e72a97
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/lora.py
View File

@ -328,6 +328,13 @@ def model_lora_keys_unet(model, key_map={}):
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
key_map[key_lora] = to
# TwinFlow-Z-Image LoRAs can target t_embedder_2.* keys.
# Alias them back to t_embedder.* targets for compatibility.
if isinstance(model, comfy.model_base.TwinFlow_Z_Image):
for key in list(key_map.keys()):
if "t_embedder." in key and "t_embedder_2." not in key:
key_map[key.replace("t_embedder.", "t_embedder_2.", 1)] = key_map[key]
if isinstance(model, comfy.model_base.Kandinsky5):
for k in sdk:
if k.startswith("diffusion_model.") and k.endswith(".weight"):