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

Prevent overwriting existing keys in state_dict

Browse Source 
Ensure new key is added only if it doesn't exist.
This commit is contained in:
azazeal04 2026-04-04 19:45:05 +02:00 committed by GitHub
parent a73ac772b5
commit 479536e34d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/lora_convert.py
View File

@ -37,7 +37,8 @@ def twinflow_z_image_lora_to_diffusers(state_dict):
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)
if new_key not in state_dict:
state_dict[new_key] = state_dict.pop(key)
return state_dict
def convert_lora(sd):