xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-03-16 10:17:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 6518bc2c0abf4202a28b06e0ab2835c44daf5daf into 650e716dda0a966a083f0efe299f3e83336f920e

Browse Source
This commit is contained in:
woctordho 2025-12-24 08:52:03 +03:00 committed by GitHub
commit b93b9d0e60
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/lora.py
View File

@ -322,6 +322,10 @@ def model_lora_keys_unet(model, key_map={}):
key_map["diffusion_model.{}".format(key_lora)] = to
key_map["transformer.{}".format(key_lora)] = to
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
for k in sdk:
if k.startswith("diffusion_model.") and k.endswith(".weight"):
key_lora = k[len("diffusion_model."):-len(".weight")]
key_map["base_model.model.{}".format(key_lora)] = k # NewBie LoRA trainer
if isinstance(model, comfy.model_base.Kandinsky5):
for k in sdk: