wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-03 13:52:31 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix key mapping for t_embedder compatibility

Browse Source 
Ensure compatibility by mapping keys for t_embedder.
This commit is contained in:
azazeal04 2026-04-04 19:42:55 +02:00 committed by GitHub
parent 5f041f4e5e
commit a73ac772b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/lora.py
View File

@ -333,7 +333,9 @@ def model_lora_keys_unet(model, key_map={}):
if isinstance(model, comfy.model_base.TwinFlow_Z_Image): if isinstance(model, comfy.model_base.TwinFlow_Z_Image):
for key in list(key_map.keys()): for key in list(key_map.keys()):
if "t_embedder." in key and "t_embedder_2." not in key: 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] key_2 = key.replace("t_embedder.", "t_embedder_2.", 1)
if key_2 not in key_map:
key_map[key_2] = key_map[key]
if isinstance(model, comfy.model_base.Kandinsky5): if isinstance(model, comfy.model_base.Kandinsky5):
for k in sdk: for k in sdk: